DeepSpeed icon indicating copy to clipboard operation
DeepSpeed copied to clipboard

Switch to adding AMD/ROCm pre-compile too

Open loadams opened this issue 1 year ago • 1 comments

  • Switch name/description of the pre-compile-ops test to include rocm in addition to cuda.
  • Use a docker image from rocm that contains rocm, python, and torch.

loadams avatar Apr 21 '23 22:04 loadams

For some reason the build can show as passing but only the cuda job passes, the rocm/rocr one fails.

loadams avatar Apr 25 '23 13:04 loadams