| .. |
|
test_attention_fallback.py
|
Fix SDPA KV-cache decode to respect sliding window (#456)
|
2026-01-30 17:32:12 +00:00 |
|
test_checkpoint_manager_new.py
|
adding unit tests, starting point of RTX 5090 optimization with flex_attention
|
2026-03-04 10:57:43 -08:00 |
|
test_common_new.py
|
adding unit tests, starting point of RTX 5090 optimization with flex_attention
|
2026-03-04 10:57:43 -08:00 |
|
test_core_eval_new.py
|
adding unit tests, starting point of RTX 5090 optimization with flex_attention
|
2026-03-04 10:57:43 -08:00 |
|
test_dataset_dataloader_new.py
|
adding unit tests, starting point of RTX 5090 optimization with flex_attention
|
2026-03-04 10:57:43 -08:00 |
|
test_engine_extra_new.py
|
adding unit tests, starting point of RTX 5090 optimization with flex_attention
|
2026-03-04 10:57:43 -08:00 |
|
test_engine.py
|
Fix MockModel's device definition (#535)
|
2026-02-17 16:03:46 -08:00 |
|
test_execution_new.py
|
adding unit tests, starting point of RTX 5090 optimization with flex_attention
|
2026-03-04 10:57:43 -08:00 |
|
test_flash_attention_new.py
|
fallback to flex_attention when FA3 is not available
|
2026-03-04 12:21:13 -08:00 |
|
test_fp8_new.py
|
adding unit tests, starting point of RTX 5090 optimization with flex_attention
|
2026-03-04 10:57:43 -08:00 |
|
test_gpt_new.py
|
adding unit tests, starting point of RTX 5090 optimization with flex_attention
|
2026-03-04 10:57:43 -08:00 |
|
test_loss_eval_new.py
|
adding unit tests, starting point of RTX 5090 optimization with flex_attention
|
2026-03-04 10:57:43 -08:00 |
|
test_optim_new.py
|
adding unit tests, starting point of RTX 5090 optimization with flex_attention
|
2026-03-04 10:57:43 -08:00 |
|
test_report_new.py
|
adding unit tests, starting point of RTX 5090 optimization with flex_attention
|
2026-03-04 10:57:43 -08:00 |
|
test_tokenizer_new.py
|
adding unit tests, starting point of RTX 5090 optimization with flex_attention
|
2026-03-04 10:57:43 -08:00 |
