[Apache TVM Discuss] [Development] Sparse OpenCL error: scheduling sparse computations that use tir.ir_builder

2021-08-07 Thread zhang677 via Apache TVM Discuss


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](https://github.com/amazon-research/FeatGraph/blob/master/python/featgraph/op/vanilla_spmm.py),
 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](https://tvm.apache.org/docs/tutorials/optimize/opt_conv_cuda.html#sphx-glr-tutorials-optimize-opt-conv-cuda-py)
 to achieve higher performance? I've also read the 
[code](https://github.com/apache/tvm/blob/main/python/tvm/topi/cuda/sparse.py#L134)
 you mentioned above, and I'm wondering if there is a shorted way to express 
such operation. Thanks!





---
[Visit 
Topic](https://discuss.tvm.apache.org/t/sparse-opencl-error-scheduling-sparse-computations-that-use-tir-ir-builder/8585/8)
 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/48c2d306f825ad308695b3cd7c36a681756201ce077f8f898ffd8d447a9b842a).


[Apache TVM Discuss] [Development] Sparse OpenCL error: scheduling sparse computations that use tir.ir_builder

2021-08-14 Thread zhang677 via Apache TVM Discuss


Thanks! Sorry for the late reply. I haven't found the workload that slow down 
the current kernel yet:).
What's more, I notice that [AutoTIR](https://github.com/apache/tvm/issues/8473) 
has been merged to main recently. Since I haven't found the doc for it, I'm 
curious about its function. And to be specific, is there any optimization for 
the hand-written kernels?





---
[Visit 
Topic](https://discuss.tvm.apache.org/t/sparse-opencl-error-scheduling-sparse-computations-that-use-tir-ir-builder/8585/10)
 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/88805fec8f15bdbc5005b28112b9a6c2f4f1a6f710b54141e93fd72998991396).


[Apache TVM Discuss] [Development] Sparse OpenCL error: scheduling sparse computations that use tir.ir_builder

2021-08-18 Thread zhang677 via Apache TVM Discuss


Got it ! Thanks for your reply!





---
[Visit 
Topic](https://discuss.tvm.apache.org/t/sparse-opencl-error-scheduling-sparse-computations-that-use-tir-ir-builder/8585/12)
 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/299bfd3dc32f6a4119d3ba3f352b8b1db02525b8f840d9eb27325c4620d9b291).