taco
taco copied to clipboard
The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
I'm trying to benchmark PyTaco in various scenarios, and it would be most helpful to have viable PyPI releases per gh-525. For example, when trying to compile from source on...
Hi, I want to transpose a sparse matrix of CSR format, where the result's format is CSC. Here is the code snippet I wrote in cpp. ```cpp int n =...
hi, i was playing around with the precompute scheduling command, but for this example http://tensor-compiler.org/codegen.html?expr=q(i,%20j)%20=%20A(i,%20j)%20*%20B(i)&format=q:ss:0,1;A:ss:0,1;B:s:0&sched=precompute:A(i,%20j)%20*%20B(i):i:i the comput function uses an `iworkspace` variable which is not defined anywhere. Is there a...
Hello, I wonder whether TACO can support expressions like `y[i] = A[i, j] * (x[j] - x[i]) ** 2`. Thanks!