kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-13 22:18:36 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[https://nvbugs/5501820][fix] Add requirements for numba-cuda version to WAR mem corruption (#7992)

Browse Source 
Signed-off-by: Pengbo Wang <221450789+pengbowang-nv@users.noreply.github.com>
This commit is contained in:
Pengbo Wang 2025-10-10 10:18:27 +08:00 committed by GitHub
parent 7b6803b6e9
commit 7da4b05289
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
requirements.txt
View File

@ -75,3 +75,4 @@ tiktoken
blobfile
openai-harmony==0.0.4
nvidia-cutlass-dsl==4.2.1; python_version >= "3.10"
numba-cuda>=0.19.0 # WAR for nvbugs/5501820

2
tests/unittest/_torch/multi_gpu/test_linear.py
View File

@ -22,8 +22,6 @@ MPI.pickle.__init__(
# needed since we reuse the mpi executor pool, first test running will leak a thread
pytestmark = pytest.mark.threadleak(enabled=False)
pytest.skip(reason="https://nvbugspro.nvidia.com/bug/5501820",
allow_module_level=True)
def rms_norm(x: torch.Tensor, weight: torch.Tensor = None, eps: float = 1e-6):