OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
gh/HDCharles/42/base
pytorch/caffe2/utils/math
History
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 [1/n]support double for Caffe2 ScatterWeightedSum (#60402) 2021-06-29 14:17:04 -07:00
elementwise.cu [ROCm] Changes not to rely on CUDA_VERSION or HIP_VERSION (#65610) 2021-09-29 09:55:43 -07:00
elementwise.h Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
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 [dte] scaffolding for c2 operator broadcasting fastpath (1/x) (#62369) 2021-07-29 16:31:02 -07:00
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 Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00
utils.cc Make PyTorch code-base clang-tidy compliant (#56892) 2021-04-28 14:10:25 -07:00
utils.h Renaming CAFFE2_API to TORCH_API (#49496) 2020-12-18 10:54:50 -08:00