pytorch/torch/utils/hipify
Xiaodong Wang eea1f79a1d [AMD] use rccl.h instead of rccl/rccl.h (#135472)
Summary: We hipify NCCLUtils.h from nccl.h to rccl/rccl.h. This follows the format of the rocm rpm suite (the header is in include/rccl/rccl.h), however the source code is just src/rccl.h. Using the rccl/rccl.h will make us find the rpm's header but not the src code's header.

Test Plan:
buck run mode/opt-amd-gpu -c hpc_comms.use_rccl=develop -c fbcode.split-dwarf=True  --config rccl.build_rdma_core=true --config rccl.adhoc_brcm=true  //aps_models/ads/icvr:icvr_launcher -- mode=local_ctr_cvr_cmf_rep_1000x_v1_no_atom   data_loader.dataset.table_ds=[2024-09-04]   data_loader.dataset.batch_size=512  max_ind_range=10

w/o this diff, it'll show 2.18 nccl version

Differential Revision: D62371434

Pull Request resolved: https://github.com/pytorch/pytorch/pull/135472
Approved by: https://github.com/jeffdaily, https://github.com/cenzhaometa
2024-10-10 08:55:57 +00:00
..
__init__.py
constants.py Revert "[ROCm] remove caffe2 from hipify (#137157)" 2024-10-08 17:45:45 +00:00
cuda_to_hip_mappings.py [AMD] use rccl.h instead of rccl/rccl.h (#135472) 2024-10-10 08:55:57 +00:00
hipify_python.py Revert "[ROCm] remove caffe2 from hipify (#137157)" 2024-10-08 17:45:45 +00:00
version.py