diff --git a/scripts/configure_pyproject_toml.py b/scripts/configure_pyproject_toml.py index 05f155fc1..958e99a60 100755 --- 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 += [