Merge pull request #18071 from hawkinsp:jitlink

PiperOrigin-RevId: 572702571
google · Oct 11, 2023 · 9ae5a43 · 9ae5a43
2 parents a1c4dc0 + 927a182
commit 9ae5a43
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -144,7 +144,7 @@ def generate_proto(source):
           # problems (https://github.com/google/jax/issues/18027#issuecomment-1756305196)
           # Until NVIDIA add version constraints, add an version constraint
           # here.
-          "nvidia-nvjitlink=cu12>=12.2",
+          "nvidia-nvjitlink-cu12>=12.2",
         ],
 
         # Target that does not depend on the CUDA pip wheels, for those who want