search

ROCm / apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
BSD 3-Clause "New" or "Revised" License
17 stars 14 forks source link

remove HCC references #122

Closed jeffdaily closed 9 months ago

jeffdaily commented 10 months ago

rename __HIP_PLATFORM_HCC__ to USE_ROCM. We were using the former to mean USE_ROCM as it is used in pytorch. The -DUSE_ROCM comes from using the pytorch CUDAExtension class.

jeffdaily commented 10 months ago

CI errors such as 

/apex/csrc/mlp_hip.hip:89:7: error: use of undeclared identifier 'rocblas_datatype_f64_r'
      rocblas_datatype_f64_r,
pruthvistony commented 9 months ago

Need to cherry-pick this patch into release branches.

@ramcherukuri , Can you please do the cherry-picks.