Addition of platform-specific key-value options to StreamExecutor DeviceOptions.

Change: 144996090

1 files changed, 6 insertions, 1 deletions

diff --git a/tensorflow/stream_executor/device_options.h b/tensorflow/stream_executor/device_options.h
index fb701790b1..b6ef800936 100644
--- a/tensorflow/stream_executor/device_options.h
+++ b/tensorflow/stream_executor/device_options.h
@@ -20,8 +20,9 @@ limitations under the License.
 #ifndef TENSORFLOW_STREAM_EXECUTOR_DEVICE_OPTIONS_H_
 #define TENSORFLOW_STREAM_EXECUTOR_DEVICE_OPTIONS_H_
 
-#include "tensorflow/stream_executor/platform/port.h"
+#include <unordered_map>
 
+#include "tensorflow/stream_executor/platform/port.h"
 #include "tensorflow/stream_executor/platform/logging.h"
 
 namespace perftools {
@@ -75,6 +76,10 @@ struct DeviceOptions {
     return flags_ == 0 ? "none" : "kDoNotReclaimStackAllocation";
   }
 
+  // Platform-specific device options. Expressed as key-value pairs to avoid
+  // DeviceOptions subclass proliferation.
+  std::unordered_map<string, string> non_portable_tags;
+
  private:
   unsigned flags_;
 };