Skip to content

Commit b50566a

Browse files
pengbowang-nvmikeiovine
authored andcommitted
[https://nvbugs/5501820][fix] Add requirements for numba-cuda version to WAR mem corruption (#7992) (#8414)
Signed-off-by: Pengbo Wang <221450789+pengbowang-nv@users.noreply.github.com> Signed-off-by: Mike Iovine <6158008+mikeiovine@users.noreply.github.com>
1 parent 95b7eab commit b50566a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

requirements.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,3 +78,5 @@ nvidia-cutlass-dsl==4.2.1; python_version >= "3.10"
7878
numba-cuda>=0.19.0 # WAR for nvbugs/5501820
7979
plotly
8080
numexpr<2.14.0 # WAR for attempted use of nonexistent numpy.typing
81+
# numba-cuda # <For CUDA 12.9>
82+
numba-cuda>=0.19.0 # WAR for nvbugs/5501820

0 commit comments

Comments
 (0)