Add missing -DGOOGLE_CUDA to dsoloader.

PiperOrigin-RevId: 249238691
bmzhao · May 21, 2019 · eae5d0b · eae5d0b
1 parent 6ad9543
commit eae5d0b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tensorflow/stream_executor/platform/default/BUILD b/tensorflow/stream_executor/platform/default/BUILD
@@ -3,6 +3,7 @@ licenses(["notice"])  # Apache 2.0
 package(default_visibility = ["//tensorflow/stream_executor:__subpackages__"])
 
 load("//tensorflow/core:platform/default/build_config_root.bzl", "if_static")
+load("//tensorflow:tensorflow.bzl", "tf_copts")
 
 cc_library(
     name = "platform",
@@ -19,6 +20,7 @@ cc_library(
         ["dlopen_checker.cc"],
     ),
     hdrs = ["dso_loader.h"],
+    copts = tf_copts(),
     deps = [
         "//tensorflow/stream_executor:platform",
         "//tensorflow/stream_executor/lib",