ROCm 6.0 prep changes (#4537)

Commit

2 years ago

ROCm 6.0 prep changes (#4537) * ROCm 6.0 prep changes * PR feedback * Try updating apex