diff options
Diffstat (limited to 'tensorflow/stream_executor/cuda/cuda_rng.cc')
-rw-r--r-- | tensorflow/stream_executor/cuda/cuda_rng.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/stream_executor/cuda/cuda_rng.cc b/tensorflow/stream_executor/cuda/cuda_rng.cc index 334c6af970..367eba4d51 100644 --- a/tensorflow/stream_executor/cuda/cuda_rng.cc +++ b/tensorflow/stream_executor/cuda/cuda_rng.cc @@ -28,7 +28,7 @@ limitations under the License. #include "tensorflow/stream_executor/lib/status.h" #include "tensorflow/stream_executor/platform/logging.h" #include "tensorflow/stream_executor/rng.h" -#include "third_party/gpus/cuda/include/curand.h" +#include "cuda/include/curand.h" // Formats curandStatus_t to output prettified values into a log stream. std::ostream &operator<<(std::ostream &in, const curandStatus_t &status) { |