Skip to content

Commit 0dce2af

Browse files
jeffdailypytorchmergebot
authored andcommitted
[ROCm][CI] adjust tf32 tolerance for test_compile_kernel_advanced (pytorch#163783)
Fixes #ISSUE_NUMBER Pull Request resolved: pytorch#163783 Approved by: https://github.com/jeffdaily Co-authored-by: Jeff Daily <[email protected]>
1 parent 71eec6a commit 0dce2af

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_cuda.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6923,7 +6923,7 @@ def test_compile_kernel_large_shared_memory(self):
69236923
with self.assertRaises(RuntimeError):
69246924
kernel.set_shared_memory_config(excessive_shared_mem)
69256925

6926-
@tf32_on_and_off(0.005)
6926+
@tf32_on_and_off(0.05 if TEST_WITH_ROCM else 0.005)
69276927
@unittest.skipIf(not TEST_CUDA, "No CUDA")
69286928
def test_compile_kernel_advanced(self):
69296929
# Test matrix multiplication

0 commit comments

Comments
 (0)