Fixes for Blackwell Tests (#1434)

paul841029 · web-flow · commit 04534c717738 · 2025-08-10T03:28:38.000-07:00
diff --git a/scripts/task_test_blackwell_kernels.sh b/scripts/task_test_blackwell_kernels.sh
@@ -7,6 +7,7 @@ set -x
 
 pip install -e . -v
 pip install --upgrade nvidia-cudnn-cu12
+pip install --upgrade cuda-python==12.*
 
 # run task_blackwell_utils_kernels.sh
 bash scripts/run_test_blackwell_utils_kernels.sh
