mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 00:20:18 +01:00
[ROCm][tunableop] Fixes flaky test issue (#166084)
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:
parent
a04edcb27a
commit
3f69b4d9b4
|
|
@ -5903,8 +5903,8 @@ class TestLinalg(TestCase):
|
|||
self.assertGreater(initial_count, 0) # we seeded 1 result line
|
||||
|
||||
# Perform ONE simple matmul
|
||||
A = torch.randn(37, 53, device=device, dtype=dtype)
|
||||
B = torch.randn(53, 29, device=device, dtype=dtype)
|
||||
A = torch.randn(27, 43, device=device, dtype=dtype)
|
||||
B = torch.randn(43, 39, device=device, dtype=dtype)
|
||||
_ = torch.matmul(A, B)
|
||||
|
||||
# Verify that new results were appended to the same file
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user