Fix for DeepSpeed build error on ROCm 5.0 and above

ROCm / DeepSpeed

DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.

https://www.deepspeed.ai/

Apache License 2.0

5 stars 3 forks source link

Fix for DeepSpeed build error on ROCm 5.0 and above #44

Closed rraminen closed 3 years ago

rraminen commented 3 years ago

The error fatal error: 'hip/hcc_detail/hip_runtime_api.h' file not found dring the DeepSPeed build for ROCm versions 5.0 and above has been fixed

cc: @jithunnair-amd

jithunnair-amd commented 3 years ago

Looks okay, if the intent is to enable the conditional code only for ROCm versions < 4.4. It seems this actually corrects a bug in the implementation of PR #36.

rraminen commented 3 years ago

DeepSpeed build log from ROCm5.0 docker DeepSpeed_build.log