oneMKL
oneMKL copied to clipboard
[BLAS][HIP] Fix blas support for rocBLAS 4+
Description
Patch is similar to https://github.com/oneapi-src/oneMKL/pull/448 (thanks @nilsfriess ) but with only the ifdef part to make the rocBLAS 4 build work.
I've tested with ROCm 5.4.3 (rocBLAS 2), and ROCm 6.1.0 (rocBLAS 4), I've also tested the build with ROCm 5.7.1 (rocBLAS 3), the new interface seemed to cause build issues with rocBLAS 3 so I kept the legacy one.
Information on the deprecated interface here.
Checklist
All Submissions
- [x] Do all unit tests pass locally? Attach a log.
- [x] Have you formatted the code using clang-format?
cc @mmeterel @Rbiessy