Hi, I’ve been working on generating CUDA code for sparse matrix multiplication using TVM in order to save me from the laborious work of designing and tuning the kernel. Since I’m new to TVM , I can’t figure out why topi cannot express the computation. In this code, it seems that the compute and schedule method can also be used in the SpMM. I suppose that maybe I can also use cache_read or other techniques described in this tutorial to achieve higher performance? I’ve also read the code you mentioned above, and I’m wondering if there is a shorted way to express such operation. Thanks!