OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[CUDA][TF32] Account for TF32 in test_alexnet_prefix (#150970)

Browse Source 
Mainly seems to be an issue on Blackwell with e.g.,
```
Mismatched elements: 1 / 746496 (0.0%)
Greatest absolute difference: 0.005461275577545166 at index (2, 32, 11, 9)
```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/150970
Approved by: https://github.com/soulitzer
This commit is contained in:
eqy 2025-04-11 23:13:54 +00:00 committed by PyTorch MergeBot
parent 2d187bf7e6
commit c4f826d5e8
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
test/inductor/test_torchinductor.py
View File

@ -5206,6 +5206,7 @@ class CommonTemplate:
check_lowp=not is_halide_backend(self.device), # misaligned addr fp16
)
@tf32_on_and_off(0.006)
@skip_if_gpu_halide # slow
def test_alexnet_prefix(self):
def forward(arg6, arg7, arg16):