Got different outputs with and without torch.compile #125999
Labels
module: correctness (silent)
issue that returns an incorrect result silently
needs reproduction
Someone else needs to try reproducing the issue given the instructions. No action needed from user
oncall: pt2
馃悰 Describe the bug
I got different results with and without @torch.compile.
If nothing goes wrong, the
a
printed the first time and the second time should be different, but they were the same when I ran it.I found when any of the following conditions is met, the first
a
and the seconda
will be different (which is what I want):@torch.compile
a
andb
are cpu tensorsa
is a multiple of 4I guess there is something wrong with the torch.compile when the last dimension of
a
is not a multiple of 4, causing this function to not be executed. I didn't get any warning or error message.I'm using A100 & H100.
Versions
cc @ezyang @gchanan @zou3519 @kadeng @msaroufim @bdhirsh @anijain2305 @chauhang
The text was updated successfully, but these errors were encountered: