ROCm / rocWMMA

rocWMMA
https://rocm.docs.amd.com/projects/rocWMMA/
MIT License
91 stars 26 forks source link

Question: can I use rocWMMA to leverage Matrix Core on rocm5.7.1? #426

Closed tiandi111 closed 2 months ago

tiandi111 commented 2 months ago

Problem Description

As described in the title, can I use rocWMMA to leverage Matrix Core on rocm5.7.1?

If not, is there any way to use Matrix Core in my own hand written kernel on rocm5.7.1?

Operating System

Ubuntu 22.04.4 LTS (Jammy Jellyfish)

CPU

Intel(R) Xeon(R) Platinum 8352Y CPU @ 2.20GHz

GPU

AMD Instinct MI210

ROCm Version

ROCm 5.7.1

ROCm Component

No response

Steps to Reproduce

No response

(Optional for Linux users) Output of /opt/rocm/bin/rocminfo --support

No response

Additional Information

No response

cgmillette commented 2 months ago

Hi @tiandi111 , yes - please refer to the release tag here