Cyrus LeungandGitHub 0e9164b40a [mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

..

__init__.py

…

allclose_default.py

…

conftest.py

…

test_activation.py

[Misc] Add CustomOp interface for device portability (#5255)

2024-06-05 09:18:19 -07:00

test_attention_selector.py

[Bugfix]: During testing, use pytest monkeypatch for safely overriding the env var that indicates the vLLM backend (#5210)

2024-06-03 20:32:57 -07:00

test_attention.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_blocksparse_attention.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_cache.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_cutlass.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_flash_attn.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_int8_quant.py

[Kernel][Misc] Use TORCH_LIBRARY instead of PYBIND11_MODULE for custom ops (#5047)

2024-06-09 16:23:30 -04:00

test_layernorm.py

[Misc] Add CustomOp interface for device portability (#5255)

2024-06-05 09:18:19 -07:00

test_marlin_gemm.py

Marlin 24 prefill performance improvement (about 25% better on average) (#4983)

2024-05-23 02:39:27 -04:00

test_moe.py

…

test_pos_encoding.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_prefix_prefill.py

…

test_rand.py

…

test_sampler.py

…

utils.py

[Bugfix]: During testing, use pytest monkeypatch for safely overriding the env var that indicates the vLLM backend (#5210)

2024-06-03 20:32:57 -07:00