rohany / taco

The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
http://tensor-compiler.org
Other
6 stars 3 forks source link

*: support a batched schedule for SpMM that uses less memory #97

Open rohany opened 2 years ago

rohany commented 2 years ago

Superceded by #102. not closing for now because this has some changes I want to upstream into master.