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

[ROCm][tunableop] Fixes flaky test issue (#166084)

Browse Source 
Fixes #165603

Pull Request resolved: https://github.com/pytorch/pytorch/pull/166084
Approved by: https://github.com/naromero77amd, https://github.com/jeffdaily
This commit is contained in:
Sarthak Tandon 2025-10-27 18:13:26 +00:00 committed by PyTorch MergeBot
parent a04edcb27a
commit 3f69b4d9b4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_linalg.py
View File

@ -5903,8 +5903,8 @@ class TestLinalg(TestCase):
self.assertGreater(initial_count, 0) # we seeded 1 result line self.assertGreater(initial_count, 0) # we seeded 1 result line
# Perform ONE simple matmul # Perform ONE simple matmul
A = torch.randn(37, 53, device=device, dtype=dtype) A = torch.randn(27, 43, device=device, dtype=dtype)
B = torch.randn(53, 29, device=device, dtype=dtype) B = torch.randn(43, 39, device=device, dtype=dtype)
_ = torch.matmul(A, B) _ = torch.matmul(A, B)
# Verify that new results were appended to the same file # Verify that new results were appended to the same file