search

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

extend TBLIS leaf kernel strategy to use CuTensor for GPUs #129

Open rohany opened 2 years ago

rohany commented 2 years ago

Follow the example in #119