I'm working on improving the current `sparse_dense` kernels on the GPU. I have a performant implementation, but it requires modification of the input sparse matrix to make it more amenable to the GPU (I have been doing this modification at compile time as the input sparse matrix is always a static weight). I have tried adding my implementation as new strategy implementation (via `OpStrategy.add_implementation`), but I cannot get it to work because I need to modify the input matrix. Is there a way to modify the inputs in this way? If not, do I need to write a new operator that uses my kernel and then write a new pass to replace existing `sparse_dense` operations with mine?
@haichen --- [Visit Topic](https://discuss.tvm.apache.org/t/modifying-weight-parameters-in-opstrategy/7914/1) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/cb133a31cdd6e9ce571d4702eceabda5c42b8ba2925656619b8cf0141da0bca3).
