add custv cutn to dependencies

PennyLaneAI · Dec 16, 2024 · 38773b1 · 38773b1
1 parent 9e97dba
commit 38773b1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/scripts/configure_pyproject_toml.py b/scripts/configure_pyproject_toml.py
@@ -122,8 +122,11 @@ def parse_args():
         "scipy-openblas32>=0.3.26",
     ]
 
+    if backend == "lightning_gpu":
+        dependencies += ["custatevec-cu12"]
+
     if backend == "lightning_tensor":
-        dependencies += ["cutensor-cu12>=2.0.2"]
+        dependencies += ["cutensornet-cu12", "cutensor-cu12>=2.0.2"]
 
     if backend in ("lightning_gpu", "lightning_tensor"):
         dependencies += [