diff options
Diffstat (limited to 'tensorflow/stream_executor/cuda/cuda_blas.cc')
-rw-r--r-- | tensorflow/stream_executor/cuda/cuda_blas.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/stream_executor/cuda/cuda_blas.cc b/tensorflow/stream_executor/cuda/cuda_blas.cc index a9dd2953e5..e2611cd3d0 100644 --- a/tensorflow/stream_executor/cuda/cuda_blas.cc +++ b/tensorflow/stream_executor/cuda/cuda_blas.cc @@ -18,8 +18,8 @@ limitations under the License. // cuda.h). This ensures that Eigen's Half.h does not attempt to make its own // __half typedef if CUDA has already defined one (and conversely, that we do // not include <cuda_fp16.h> after Half.h has made its typedef). -#include "third_party/gpus/cuda/include/cuda.h" -#include "third_party/gpus/cuda/include/cublas_v2.h" +#include "cuda/include/cuda.h" +#include "cuda/include/cublas_v2.h" #if CUDA_VERSION >= 7050 #define EIGEN_HAS_CUDA_FP16 |