tensor-compiler / taco

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

pytaco.tensor_sum() gives a wrong answer #538

Open JoshuaQSH opened 1 year ago

JoshuaQSH commented 1 year ago

Hi guys,

I have been playing pytaco with its API recently. But while running a simple pytaco.tensor_sum(..., axis=0) , it seems to give a wrong answer:

image

Is it just me or has someone also come across this?