ROCm / MIOpen

AMD's Machine Intelligence Library
https://rocm.docs.amd.com/projects/MIOpen/en/latest/
Other
1.05k stars 219 forks source link

Simple GEMM Example with Xdlops #1091

Closed raikonenfnu closed 6 months ago

raikonenfnu commented 3 years ago

Hey guys, I am trying to set up a toy example of gemm using xdlops with the miopen, I was wondering if there are any simple examples on doing such, the closest things I have seen that give extra info is gemm.hpp and [amd_xdlops.hpp] refered by Jack Chung and (https://github.com/ROCmSoftwarePlatform/MIOpen/blob/c701af3ba83f8fb99d7efab30ee0b22de57f2f42/src/kernels/composable_kernel/include/utility/amd_xdlops.hpp).

Jack also mentioned to tag @zjing14 who is apparently the expert in this field 😄 . Thanks in advance!

raikonenfnu commented 3 years ago

My final goal is to bring up a simple MLIR based gemm that runs on xdlops in ROCm MLIR, I have seen the examples but still don't quite understand how the matrixA and matrixB is size of memref<12288xf16, 3>, even though m,n=256, and k = 16

whchung commented 3 years ago
junliume commented 3 years ago

@raikonenfnu Thanks for getting in touch. Would you be interested in a brief intro meeting so that we can understand your requests more effectively? My contact email is listed under profile, and we can start from there to try to set something up next week. And thanks to @whchung ! CC: @zjing14 we can continue discussion thread here too, in case the request looks clear to you.

raikonenfnu commented 3 years ago

@raikonenfnu Thanks for getting in touch. Would you be interested in a brief intro meeting so that we can understand your requests more effectively? My contact email is listed under profile, and we can start from there to try to set something up next week. And thanks to @whchung ! CC: @zjing14 we can continue discussion thread here too, in case the request looks clear to you.

Sounds good! will shoot you an email and set up a meeting with my boss at Nod Labs as well 😄

raikonenfnu commented 3 years ago

sent an email, my email is stanley@nod-labs.com

junliume commented 3 years ago

@raikonenfnu has the issue been resolved for you?