Mor Zusman f13a07b1f8 [Kernel][Model] Varlen prefill + Prefill chunking support for mamba kernels and Jamba model (#8533)

2024-09-29 17:35:58 -04:00

..

decoder_only

[Kernel][Model] Varlen prefill + Prefill chunking support for mamba kernels and Jamba model (#8533)

2024-09-29 17:35:58 -04:00

embedding

[CI/Build] Reorganize models tests (#7820)

2024-09-13 10:20:06 -07:00

encoder_decoder

[CI/Build] Add test decorator for minimum GPU memory (#8925)

2024-09-29 02:50:51 +00:00

fixtures

[CI/Build] Update pixtral tests to use JSON (#8436)

2024-09-13 03:47:52 +00:00

__init__.py

[CI/Build] Move test_utils.py to tests/utils.py (#4425)

2024-05-13 23:50:09 +09:00

test_oot_registration.py

[misc][ci] fix cpu test with plugins (#7489)

2024-08-13 19:27:46 -07:00

test_registry.py

[BugFix] Fix test breakages from transformers 4.45 upgrade (#8829)

2024-09-26 16:46:43 -07:00

utils.py

[CI/Build] Update models tests & examples (#8874)

2024-09-28 09:54:35 -07:00