Added ROCM_VERSION compiler directives

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

4 stars 3 forks source link

Closed rraminen closed 2 years ago

rraminen commented 2 years ago

Added ROCM_VERSION_MINOR and ROCM_VERSION_MAJOR compiler directives to enable the build latest ROCm versions.

cc: @jithunnair-amd