Skip to content

Commit 108decd

Browse files
committed
Adds nvtx dependency
1 parent 840c6e5 commit 108decd

File tree

2 files changed

+7
-6
lines changed

2 files changed

+7
-6
lines changed

.github/scripts/generate_binary_build_matrix.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,8 @@ def generate_wheels_matrix(os: str,
226226
"nvidia-cufft-cu11==10.9.0.58; platform_system == 'Linux' | "
227227
"nvidia-curand-cu11==10.2.10.91; platform_system == 'Linux' | "
228228
"nvidia-cusolver-cu11==11.4.0.1; platform_system == 'Linux' | "
229-
"nvidia-cusparse-cu11==11.7.4.91; platform_system == 'Linux'",
229+
"nvidia-cusparse-cu11==11.7.4.91; platform_system == 'Linux' | "
230+
"nvidia-nvtx-cu11==11.7.91; platform_system == 'Linux'",
230231
"build_name":
231232
f"{package_type}-py{python_version}-{gpu_arch_type}{gpu_arch_version}-with-pypi-cudnn"
232233
.replace(

.github/workflows/generated-linux-binary-manywheel-nightly.yml

Lines changed: 5 additions & 5 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)