diff options
author | Nick Desaulniers <ndesaulniers@google.com> | 2018-05-17 14:22:49 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-17 14:25:26 -0700 |
commit | 2b03e28d480714f42b950468752fb6a99030fcf8 (patch) | |
tree | 3ef119680eac438856d5fcd2eb183d6d601d1f2d /tensorflow/stream_executor/multi_platform_manager.h | |
parent | 8ab1bc703256928bbb570e33e2693a04ef5eda60 (diff) |
[TF:STREAM_EXECUTOR] move declaration of multi_platform_manager to header
PiperOrigin-RevId: 197051272
Diffstat (limited to 'tensorflow/stream_executor/multi_platform_manager.h')
-rw-r--r-- | tensorflow/stream_executor/multi_platform_manager.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/multi_platform_manager.h b/tensorflow/stream_executor/multi_platform_manager.h index 7e316879ca..146a128e85 100644 --- a/tensorflow/stream_executor/multi_platform_manager.h +++ b/tensorflow/stream_executor/multi_platform_manager.h @@ -68,6 +68,7 @@ limitations under the License. #include <map> #include <memory> +#include "tensorflow/stream_executor/lib/initialize.h" #include "tensorflow/stream_executor/lib/status.h" #include "tensorflow/stream_executor/lib/statusor.h" #include "tensorflow/stream_executor/platform.h" @@ -182,4 +183,9 @@ class MultiPlatformManager { } // namespace stream_executor +// multi_platform_manager.cc will define this instance. Includers of this header +// should use +// REGISTER_MODULE_INITIALIZER_SEQUENCE(my_platform, multi_platform_manager); +DECLARE_MODULE_INITIALIZER(multi_platform_manager); + #endif // TENSORFLOW_STREAM_EXECUTOR_MULTI_PLATFORM_MANAGER_H_ |