oneapi-src / oneMKL

oneAPI Math Kernel Library (oneMKL) Interfaces

Apache License 2.0

624 stars 159 forks source link

[SPARSE] Add support for rocSPARSE backend #544

Open Rbiessy opened 3 months ago

Rbiessy commented 3 months ago

Description

Add support for the rocSPARSE backend.

Depends on https://github.com/oneapi-src/oneMKL/pull/527 and https://github.com/oneapi-src/oneMKL/pull/532.

Rendered docs: docs.zip

Checklist

All Submissions

[x] Do all unit tests pass locally? amd_log.txt nvidia_log.txt intel_log.txt
- Tested the rocSPARSE backend on MI210 with oneAPI 2024.2 compiler + Codeplay AMD plugin + ROCm 6.1.0
[x] Have you formatted the code using clang-format?

Rbiessy commented 3 months ago

Tests log on W6800: amd_w6800_log.txt

gajanan-choudhary commented 3 weeks ago

@Rbiessy, now that #527 is merged in, could you please rebase this branch against main, resolve conflicts, and force-push when you can so that I can start reviewing this PR?

Rbiessy commented 3 weeks ago

Yes, this is in progress! I expect I will need a few days at least. I'm aiming to merge the PR by the end of the year.

Rbiessy commented 2 weeks ago

@gajanan-choudhary I have updated the PR with recent changes from cuSPARSE. Note that it moved almost all the content from cusparse_task.hpp to common_launch_task.hpp with almost no changes. New test logs: log_mi210.txt log_a100.txt log_pvc.txt