TensorRT-LLMs/tests/unittest/_torch/multi_gpu_modeling
Jinyang Yuan f9a9a1af2e
[fix] Fix Llama4 allgather error due to None tensor (#4511)
* [fix] Fix Llama4 allgather error due to None tensor

Signed-off-by: Jinyang Yuan <154768711+jinyangyuan-nvidia@users.noreply.github.com>

* Refactor modifications

Signed-off-by: Jinyang Yuan <154768711+jinyangyuan-nvidia@users.noreply.github.com>

* Minor modification

Signed-off-by: Jinyang Yuan <154768711+jinyangyuan-nvidia@users.noreply.github.com>

* Minor fix

Signed-off-by: Jinyang Yuan <154768711+jinyangyuan-nvidia@users.noreply.github.com>

---------

Signed-off-by: Jinyang Yuan <154768711+jinyangyuan-nvidia@users.noreply.github.com>
2025-05-24 19:12:12 +08:00
..
test_deepseek.py perf: Eliminate the need for attention DP padding when possible (#3439) 2025-05-17 13:30:55 +08:00
test_llama4.py [fix] Fix Llama4 allgather error due to None tensor (#4511) 2025-05-24 19:12:12 +08:00