Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Python CUDA dependencies as Lightning-GPU and Lightning-Tensor wheel dependencies #1025

Merged
merged 36 commits into from
Jan 3, 2025
Merged

Add Python CUDA dependencies as Lightning-GPU and Lightning-Tensor wheel dependencies #1025

Show file tree
Hide file tree
Changes from 21 commits
Commits
Show all changes
36 commits
Select commit Hold shift + click to select a range
8a1b922
just gpu/tensor
josephleekl Dec 12, 2024
be99017
only change configure script
josephleekl Dec 12, 2024
5b244c6
update
josephleekl Dec 12, 2024
c34fb60
update configure
josephleekl Dec 13, 2024
69e3de5
Auto update version from '0.40.0-dev35' to '0.40.0-dev36'
ringo-but-quantum Dec 13, 2024
ccda2b2
add to build requirements
josephleekl Dec 13, 2024
3b1e514
Merge branch 'cuda-wheel-dep' of github.com:PennyLaneAI/pennylane-lig…
josephleekl Dec 13, 2024
324a355
update
josephleekl Dec 13, 2024
3a31bc4
format
josephleekl Dec 13, 2024
729e49a
setup
josephleekl Dec 13, 2024
f597ae4
setup
josephleekl Dec 13, 2024
4211e3a
configure
josephleekl Dec 13, 2024
c249ffe
Cmakelist
josephleekl Dec 13, 2024
4ba9485
install rpaths
josephleekl Dec 13, 2024
27247a1
install rpaths
josephleekl Dec 13, 2024
a5f1be6
Auto update version from '0.40.0-dev36' to '0.40.0-dev37'
ringo-but-quantum Dec 13, 2024
a2b7cf3
remove duplicated cusparse
josephleekl Dec 13, 2024
c6aa996
Merge branch 'cuda-wheel-dep' of github.com:PennyLaneAI/pennylane-lig…
josephleekl Dec 13, 2024
09bcdc3
configure
josephleekl Dec 13, 2024
503321e
remove duplicated cusparse
josephleekl Dec 13, 2024
9e97dba
revert audit wheel
josephleekl Dec 16, 2024
38773b1
add custv cutn to dependencies
josephleekl Dec 16, 2024
cc608ee
Auto update version from '0.40.0-dev37' to '0.40.0-dev38'
ringo-but-quantum Dec 16, 2024
ca44964
Merge branch 'master' into cuda-wheel-dep
josephleekl Dec 16, 2024
1770c87
build wheels in draft mode
josephleekl Dec 16, 2024
205d21c
revert workflows and remove cusolver for lgpu
josephleekl Dec 16, 2024
fb36e6b
changelog
josephleekl Dec 16, 2024
0185289
Merge branch 'master' into cuda-wheel-dep
josephleekl Dec 19, 2024
279b2d4
add custatevec/cutensornet sys_platform linux requirement
josephleekl Dec 19, 2024
c963452
Auto update version from '0.40.0-dev41' to '0.40.0-dev42'
ringo-but-quantum Dec 19, 2024
8bc07b4
remove cutensor
josephleekl Dec 20, 2024
bb93b03
Merge branch 'cuda-wheel-dep' of github.com:PennyLaneAI/pennylane-lig…
josephleekl Dec 20, 2024
108641e
Merge branch 'master' into cuda-wheel-dep
multiphaseCFD Jan 2, 2025
e059c5b
Auto update version from '0.40.0-dev42' to '0.40.0-dev43'
ringo-but-quantum Jan 2, 2025
252e826
Trigger CIs
multiphaseCFD Jan 2, 2025
7bc739a
set mpi workflow timeout-minutes 35
multiphaseCFD Jan 2, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion pennylane_lightning/core/_version.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,4 +16,4 @@
Version number (major.minor.patch[-label])
"""

__version__ = "0.40.0-dev35"
__version__ = "0.40.0-dev37"
4 changes: 2 additions & 2 deletions pennylane_lightning/core/src/simulators/lightning_gpu/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ set_property(TARGET ${PL_BACKEND} PROPERTY POSITION_INDEPENDENT_CODE ON)

# To avoid DSO errors on platforms preferring static linkage, uncomment the following line:
# string(REPLACE "libcudart_static.a" "libcudart.so" CUDA_SHARED_RT "${CUDA_LIBRARIES}")
set_target_properties(${PL_BACKEND} PROPERTIES INSTALL_RPATH "$ORIGIN/../cuquantum/lib:$ORIGIN/../cuquantum/lib64:$ORIGIN/")
set_target_properties(${PL_BACKEND} PROPERTIES INSTALL_RPATH "$ORIGIN/../cuquantum/lib:$ORIGIN/../cuquantum/lib64:$ORIGIN/../nvidia/cublas/lib:$ORIGIN/../nvidia/cusparse/lib:$ORIGIN/../nvidia/nvjitlink/lib:$ORIGIN/../nvidia/cusolver/lib:$ORIGIN/../nvidia/cuda_runtime/lib:$ORIGIN/")

if(PL_DISABLE_CUDA_SAFETY)
target_compile_options(${PL_BACKEND} INTERFACE $<$<COMPILE_LANGUAGE:CXX>:-DCUDA_UNSAFE>)
Expand All @@ -87,4 +87,4 @@ endforeach()
if (BUILD_TESTS)
enable_testing()
add_subdirectory("tests")
endif()
endif()
2 changes: 1 addition & 1 deletion pennylane_lightning/core/src/simulators/lightning_tensor/tncuda/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ set_property(TARGET ${PL_BACKEND} PROPERTY POSITION_INDEPENDENT_CODE ON)

# To avoid DSO errors on platforms preferring static linkage, uncomment the following line:
# string(REPLACE "libcudart_static.a" "libcudart.so" CUDA_SHARED_RT "${CUDA_LIBRARIES}")
set_target_properties(${PL_BACKEND} PROPERTIES INSTALL_RPATH "$ORIGIN/../cuquantum/lib:$ORIGIN/../cuquantum/lib64:$ORIGIN/")
set_target_properties(${PL_BACKEND} PROPERTIES INSTALL_RPATH "$ORIGIN/../cuquantum/lib:$ORIGIN/../cuquantum/lib64:$ORIGIN/../nvidia/cublas/lib:$ORIGIN/../nvidia/cusparse/lib:$ORIGIN/../cutensor/lib:$ORIGIN/../nvidia/nvjitlink/lib:$ORIGIN/../nvidia/cusolver/lib:$ORIGIN/../nvidia/cuda_runtime/lib:$ORIGIN/")

if(PL_DISABLE_CUDA_SAFETY)
target_compile_options(${PL_BACKEND} INTERFACE $<$<COMPILE_LANGUAGE:CXX>:-DCUDA_UNSAFE>)
Expand Down
12 changes: 12 additions & 0 deletions scripts/configure_pyproject_toml.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -122,6 +122,18 @@ def parse_args():
"scipy-openblas32>=0.3.26",
]

if backend == "lightning_tensor":
dependencies += ["cutensor-cu12>=2.0.2"]
josephleekl marked this conversation as resolved.
Show resolved Hide resolved

if backend in ("lightning_gpu", "lightning_tensor"):
dependencies += [
josephleekl marked this conversation as resolved.
Show resolved Hide resolved
josephleekl marked this conversation as resolved.
Show resolved Hide resolved
"nvidia-nvjitlink-cu12",
"nvidia-cusparse-cu12",
"nvidia-cusolver-cu12",
"nvidia-cublas-cu12",
"nvidia-cuda-runtime-cu12",
]

if backend != "lightning_qubit":
dependencies += ["pennylane_lightning==" + version]

Expand Down
Loading