[Tracking Issue] Support any WMMA shape in rgcn/sparse conv

uwsampl / SparseTIR

SparseTIR: Sparse Tensor Compiler for Deep Learning

https://sampl.cs.washington.edu/SparseTIR/

Apache License 2.0

131 stars 14 forks source link

[Tracking Issue] Support any WMMA shape in rgcn/sparse conv #63

Open yzh119 opened 2 years ago

yzh119 commented 2 years ago

Currently, we only use m16n16k16 wmma instructions in RGCN/Sparse Conv which might not be optimal, we should refactor the code to enable different WMMA shapes (m8n32k16/m32n8k16/m16n16k16).