Xiang Gao b8dfb45ac2 Refactor cub namespace handling (#66219) ... Summary: This PR is to update PyTorch with the following cub changes: - Starting cub 1.13.1, cub requires users to define `CUB_NS_QUALIFIER` if `CUB_NS_PREFIX` is also defined. Besides that, a new mechanism `CUB_WRAPPED_NAMESPACE` is added. And I do the following change to PyTorch: - Starting CUDA 11.5, define `CUB_WRAPPED_NAMESPACE` globally as an nvcc flag. - Fix caffe2 failures caused by the above change. - Add a `aten/src/ATen/cuda/cub_definitions.cuh` that defines helper macros about feature availability. Pull Request resolved: https://github.com/pytorch/pytorch/pull/66219 Reviewed By: bdhirsh Differential Revision: D31626931 Pulled By: ngimel fbshipit-source-id: 97ebf5ef671ade8bf46d0860edc317f22660f26d		2021-10-25 14:37:09 -07:00
..
broadcast.cc	[dte] broadcast fastpath implementations for reduce utility functions (2/x) (#62428)	2021-07-29 17:27:39 -07:00
broadcast.cu	[ROCm] Changes not to rely on CUDA_VERSION or HIP_VERSION (#65610)	2021-09-29 09:55:43 -07:00
broadcast.h	[dte] broadcast fastpath implementations for reduce utility functions (2/x) (#62428)	2021-07-29 17:27:39 -07:00
elementwise.cc
elementwise.cu	[ROCm] Changes not to rely on CUDA_VERSION or HIP_VERSION (#65610)	2021-09-29 09:55:43 -07:00
elementwise.h
half_utils.h
reduce.cc	[dte] broadcast fastpath implementations for reduce utility functions (2/x) (#62428)	2021-07-29 17:27:39 -07:00
reduce.cu	Refactor cub namespace handling (#66219)	2021-10-25 14:37:09 -07:00
reduce.cuh	Refactor cub namespace handling (#66219)	2021-10-25 14:37:09 -07:00
reduce.h
transpose.cc
transpose.cu	[ROCm] Changes not to rely on CUDA_VERSION or HIP_VERSION (#65610)	2021-09-29 09:55:43 -07:00
transpose.h
utils.cc
utils.h