relax tolerance

pytorch · May 14, 2024 · a8b4dd3 · a8b4dd3
1 parent 08a95ac
commit a8b4dd3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_ops.py b/test/test_ops.py
@@ -133,7 +133,7 @@ def test_fp6_matmul_correctness(self, BS, OC, IC, splitK):
 
         error = (results_fp6 - results_fp16).abs()
         relative_error = error / results_fp16.abs()
-        assert relative_error.mean() < 1e-3
+        assert relative_error.mean() < 1e-2
 
 
 if __name__ == "__main__":