jon-chuang 5231d57c71 [TESTS] replace deprecated torch.testing.assert_allclose (#2250) ...

Prior to this PR, matmul on sm_89 (RTX 4070)
(`test/unit/operators/test_matmul.py::test_op`) would result in test
failure due to too strict atol/rtol.

To avoid having to choose strictness ourselves, and to have better
defaults based on dtype, use the non-deprecated torch testing util.

See: https://github.com/pytorch/pytorch/issues/61844

Replace: https://github.com/openai/triton/pull/2242

2023-09-11 15:31:17 -04:00

..

ttgir_tests

[TESTS] remove get_proper_err, get_variant_golden (#2039)

2023-08-07 22:52:55 -07:00

__init__.py

Initial code merge of Hopper support (#2036)

2023-08-07 09:53:04 +08:00

test_flashattention.py

Initial code merge of Hopper support (#2036)

2023-08-07 09:53:04 +08:00

test_gemm_fusion.py

Initial code merge of Hopper support (#2036)

2023-08-07 09:53:04 +08:00

test_gemm.py

[TESTS][HOPPER] Prune hopper tests to speedup CI (#2193)

2023-08-27 20:45:23 -07:00

test_mixed_io.py

Initial code merge of Hopper support (#2036)

2023-08-07 09:53:04 +08:00

test_persistent_warp_specialized_fused-attention.py

[TESTS] refactor test-persistent-warp-specialized-gemm UTs (#2075)

2023-08-10 06:57:04 +00:00

test_persistent_warp_specialized_gemm.py

[TESTS] replace deprecated torch.testing.assert_allclose (#2250)

2023-09-11 15:31:17 -04:00

test_tma_store_gemm.py

[TESTS] remove get_proper_err, get_variant_golden (#2039)

2023-08-07 22:52:55 -07:00