diff options
author | Gunhan Gulsoy <gunan@google.com> | 2018-02-09 14:37:24 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-09 14:44:16 -0800 |
commit | 7e23850c2145ed565c668d6ba327dbcf064d4ed8 (patch) | |
tree | 3edbb0471cbeb0439aa36c8363b265150b392ef5 /tensorflow/stream_executor/dso_loader.cc | |
parent | dc1e7b8a635426db809a094b313cc6be40ab9276 (diff) |
Remove header dependence on cuda_config.h to fix opensource custom op support.
Fixes #14454
Fixes #12860
PiperOrigin-RevId: 185194924
Diffstat (limited to 'tensorflow/stream_executor/dso_loader.cc')
-rw-r--r-- | tensorflow/stream_executor/dso_loader.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/dso_loader.cc b/tensorflow/stream_executor/dso_loader.cc index 0c642912b1..9516883627 100644 --- a/tensorflow/stream_executor/dso_loader.cc +++ b/tensorflow/stream_executor/dso_loader.cc @@ -33,6 +33,10 @@ limitations under the License. #include "tensorflow/stream_executor/platform/logging.h" #include "tensorflow/stream_executor/platform/port.h" +#if !defined(PLATFORM_GOOGLE) +#include "cuda/cuda_config.h" +#endif + namespace perftools { namespace gputools { namespace internal { |