Dongdong Li 0469d5fccd [OPTIMIZER] Remove extra wgmma_wait_group in flash attention (#2399) ...

Co-authored-by: dongdongl <dongdongl@nvidia.com>

2023-10-26 16:35:36 +00:00

..

test_blocksparse.py

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

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

test_cross_entropy.py

[BACKEND] Remove dependency between NVGPU and TritonNvidiaGPU (#2282)

2023-09-12 11:02:20 -07:00

test_flash_attention.py

[OPTIMIZER] Remove extra wgmma_wait_group in flash attention (#2399)

2023-10-26 16:35:36 +00:00

test_inductor.py

[BACKEND] shfl ptx insts should have side effects (#2376)

2023-09-23 10:05:20 -07:00

test_matmul.py

[FRONTEND][BACKEND] Add flag to control accumulation for fp8 (#2300)

2023-09-15 18:42:54 -07:00