DeepSpeed
0daf7f2a - Added __HIP_PLATFORM_AMD__=1 for non JIT build (#4585)

Commit
2 years ago
Added __HIP_PLATFORM_AMD__=1 for non JIT build (#4585) This PR is required in addition to https://github.com/microsoft/DeepSpeed/pull/4539 to define HIP_PLATFORM_AMD on ROCm. This is required for DeepSpeed non-JIT build. For JIT build we have https://github.com/microsoft/DeepSpeed/pull/4570. This is required for DeepSpeed extensions build in those docker images with PyTorch built before https://github.com/pytorch/pytorch/pull/111975. cc: @jeffdaily @jithunnair-amd Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
Author
Parents
Loading