diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-20 23:31:26 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-20 23:36:01 -0700 |
commit | cf047f7755f3400ee128db2571042091fe9f8314 (patch) | |
tree | c60f49647393fa838cc2b9e04464a961e6083b37 /tensorflow/core/kernels/reduction_gpu_kernels.cu.h | |
parent | 23552a8b2f2a92a31710b9339e6ade514ac25996 (diff) |
Fix cub include path so that TensorFlow compiles when used as a bazel dependency.
PiperOrigin-RevId: 213942340
Diffstat (limited to 'tensorflow/core/kernels/reduction_gpu_kernels.cu.h')
-rw-r--r-- | tensorflow/core/kernels/reduction_gpu_kernels.cu.h | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/tensorflow/core/kernels/reduction_gpu_kernels.cu.h b/tensorflow/core/kernels/reduction_gpu_kernels.cu.h index 88b3c2ac76..bb8254eaac 100644 --- a/tensorflow/core/kernels/reduction_gpu_kernels.cu.h +++ b/tensorflow/core/kernels/reduction_gpu_kernels.cu.h @@ -21,11 +21,11 @@ limitations under the License. #define EIGEN_USE_GPU #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" -#include "external/cub_archive/cub/device/device_reduce.cuh" -#include "external/cub_archive/cub/device/device_segmented_reduce.cuh" -#include "external/cub_archive/cub/iterator/counting_input_iterator.cuh" -#include "external/cub_archive/cub/iterator/transform_input_iterator.cuh" -#include "external/cub_archive/cub/warp/warp_reduce.cuh" +#include "third_party/cub/device/device_reduce.cuh" +#include "third_party/cub/device/device_segmented_reduce.cuh" +#include "third_party/cub/iterator/counting_input_iterator.cuh" +#include "third_party/cub/iterator/transform_input_iterator.cuh" +#include "third_party/cub/warp/warp_reduce.cuh" #include "cuda/include/cuComplex.h" #include "tensorflow/core/kernels/reduction_ops.h" #include "tensorflow/core/lib/core/bits.h" |