Pamela Peng 6cdfc54883 feat: Add FP8 support for SM 120 (#3248) ... * Allow FP8 on SM120 Signed-off-by: Pamela Peng <179191831+pamelap-nvidia@users.noreply.github.com> * fix sm121 Signed-off-by: Pamela Peng <179191831+pamelap-nvidia@users.noreply.github.com> * fix Signed-off-by: Pamela Peng <179191831+pamelap-nvidia@users.noreply.github.com> * fix pre-commit Signed-off-by: Pamela Peng <179191831+pamelap-nvidia@users.noreply.github.com> * review update Signed-off-by: Pamela Peng <179191831+pamelap-nvidia@users.noreply.github.com> --------- Signed-off-by: Pamela Peng <179191831+pamelap-nvidia@users.noreply.github.com> Co-authored-by: Sharan Chetlur <116769508+schetlur-nv@users.noreply.github.com>		2025-04-14 16:05:41 -07:00
..
test_modeling_bert.py	feat: no-cache attention in PyTorch workflow (#3085)	2025-04-05 01:54:32 +08:00
test_modeling_llama.py	test: reorganize tests folder hierarchy (#2996)	2025-03-27 12:07:53 +08:00
test_modeling_mamba_hybrid.py	test: reorganize tests folder hierarchy (#2996)	2025-03-27 12:07:53 +08:00
test_modeling_mixtral.py	feat: Add FP8 support for SM 120 (#3248)	2025-04-14 16:05:41 -07:00
test_modeling_mllama.py	test: reorganize tests folder hierarchy (#2996)	2025-03-27 12:07:53 +08:00
test_modeling_nemotron_nas.py	chore: Rename nvsmall to nemotron nas (#3447)	2025-04-10 23:16:52 +08:00
test_modeling_nemotron.py	test: reorganize tests folder hierarchy (#2996)	2025-03-27 12:07:53 +08:00
test_modeling_out_of_tree.py	Add thread leak check and fix thread/memory leak issues. (#3270)	2025-04-08 19:03:18 +08:00
test_modeling_qwen_moe.py	feat: add qwen2 moe to torch flow; fix wrong imported KvCacheConfig in gpqa… (#3369)	2025-04-10 22:45:57 +08:00
test_modeling_qwen.py	feat: no-cache attention in PyTorch workflow (#3085)	2025-04-05 01:54:32 +08:00
test_modeling_vila.py	fix vila test (#3042)	2025-04-04 14:30:06 +08:00