-
Notifications
You must be signed in to change notification settings - Fork 184
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
kernel seg fault when processing large tensors #556
Comments
TACO currently (and it's not a trivial fix) generates code using 32 bit integers. The dimension sizes of nell-1 will cause integer overflow for your dense output tensor, leading to the segfault that you're seeing. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hello, I got seg faults using the kernel generated by taco when processing large tensors. I used taco compiler to generate the following kernel computation
and use the following
main
functionI replaced the
restrict
keyword with__restrict__
in the generated two *.hpp files and compile it usingclang
from LLVM15 with the following commandThe sparse tensors I used are
nell-1
andnell-2
from FROSTT, which are also used for the original taco paper. However, the kernel successfully run with the tensornell-2
but not for the larger onenell-1
(seg fault). Is this a known issue or it's a bug such as data overflow within taco? The issue happens to other large tensors as well.The text was updated successfully, but these errors were encountered: