Skip to content

Commit 35daec9

Browse files
[release/2.7][ROCm][tunableop] UT tolerance increase for matmul_small_brute_force_… (#2397)
TunableOp will sometimes find a less precise solution due to the small input vectors used in this UT. Bumping up tolerance to eliminate flakiness. Pull Request resolved: pytorch#158788 Approved by: https://github.com/jeffdaily (cherry picked from commit c917c63)
1 parent 6b52d83 commit 35daec9

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/test_linalg.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4704,6 +4704,7 @@ def test_matmul_small_brute_force_3d_Nd(self, device, dtype):
47044704
@onlyCUDA
47054705
@skipCUDAIfNotRocm # Skipping due to SM89 OOM in CI, UT doesn't do much on NV anyways
47064706
@dtypes(*floating_types_and(torch.half))
4707+
@precisionOverride({torch.float16: 1e-1}) # TunableOp may occasionally find less precise solution
47074708
def test_matmul_small_brute_force_tunableop(self, device, dtype):
47084709
# disable tunableop buffer rotation for all tests everywhere, it can be slow
47094710
# We set the TunableOp numerical check environment variable here because it is

0 commit comments

Comments
 (0)