liji-nv b168adba70 feat: Add NVFP4 UB pattern optimization pass in torch compile (#3371) ... * feat: Add NVFP4 UB pattern optimization pass in torch compile * Add an additional flag for UB fp4 pattern to avoid inverse the scale * Add NVFP4 related UB patterns Signed-off-by: Jin Li <59594262+liji-nv@users.noreply.github.com> * Update atol, some points fails for B200 umbriel. Signed-off-by: liji-nv <59594262+liji-nv@users.noreply.github.com> --------- Signed-off-by: Jin Li <59594262+liji-nv@users.noreply.github.com> Signed-off-by: liji-nv <59594262+liji-nv@users.noreply.github.com>		2025-04-11 21:25:29 +08:00
..
test_ar_residual_norm.py	test: reorganize tests folder hierarchy (#2996)	2025-03-27 12:07:53 +08:00
test_deepseek_allreduce.py	fix: runtime error in est_deepseek_allreduce.py (#3226)	2025-04-08 09:19:47 +08:00
test_embedding.py	Update TensorRT-LLM (#2936)	2025-03-18 21:25:19 +08:00
test_linear.py	Update TensorRT-LLM (#2936)	2025-03-18 21:25:19 +08:00
test_star_attention_input.jsonl	Update TensorRT-LLM (#2936)	2025-03-18 21:25:19 +08:00
test_star_attention.py	move BuildConfig functional args to llmargs (#3036)	2025-03-29 02:20:18 +08:00
test_user_buffers.py	feat: Add NVFP4 UB pattern optimization pass in torch compile (#3371)	2025-04-11 21:25:29 +08:00