Remove header dependence on cuda_config.h to fix opensource custom op support.
authorGunhan Gulsoy <gunan@google.com>
Fri, 9 Feb 2018 22:37:24 +0000 (14:37 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 9 Feb 2018 22:44:16 +0000 (14:44 -0800)
Fixes #14454
Fixes #12860

PiperOrigin-RevId: 185194924

tensorflow/core/common_runtime/gpu/gpu_device.cc
tensorflow/stream_executor/dso_loader.cc
tensorflow/stream_executor/dso_loader.h

index 0fb908b08025c3cf98b67e82df9abd75a6400111..15ff15fd5ab28605c4ab0904e62305edc3815adb 100644 (file)
@@ -66,6 +66,10 @@ limitations under the License.
 #include "tensorflow/core/util/env_var.h"
 #include "tensorflow/core/util/stream_executor_util.h"
 
+#if !defined(PLATFORM_GOOGLE)
+#include "cuda/cuda_config.h"
+#endif
+
 namespace tensorflow {
 
 // Eigen Ops directly allocate memory only for temporary buffers used
index 0c642912b1ff06dd5754f841fe417387e3725361..95168836278add5d6592ff0c3d0f7245e6f6bc5b 100644 (file)
@@ -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 {
index 9495f7253a1d475f0b5321b71419febd086832af..354c7b50b8209755991827b3c36afac790cb952b 100644 (file)
@@ -28,10 +28,6 @@ limitations under the License.
 #include "tensorflow/stream_executor/platform.h"
 #include "tensorflow/stream_executor/platform/mutex.h"
 
-#if !defined(PLATFORM_GOOGLE)
-#include "cuda/cuda_config.h"
-#endif
-
 namespace perftools {
 namespace gputools {
 namespace internal {