Skip to content

Commit 81e858c

Browse files
authored
Compile virtual CUDA architecture for libxc (pyscf#347)
* 70-virtual + 70-real * v0.7 * install cutensor
1 parent 37bdc01 commit 81e858c

File tree

4 files changed

+6
-2
lines changed

4 files changed

+6
-2
lines changed

.github/workflows/nightly_build.yml

+3
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ jobs:
2828
pip3 install cupy-cuda12x --upgrade
2929
pip3 install gpu4pyscf-libxc-cuda12x --upgrade
3030
git config --global core.compression 9
31+
python3 -m cupyx.tools.install_library --cuda 12.x --library cutensor
3132
- name: Run RKS tests
3233
run: |
3334
export CUDA_HOME=/usr/local/cuda
@@ -52,6 +53,7 @@ jobs:
5253
pip3 install cupy-cuda12x --upgrade
5354
pip3 install gpu4pyscf-libxc-cuda12x --upgrade
5455
git config --global core.compression 9
56+
python3 -m cupyx.tools.install_library --cuda 12.x --library cutensor
5557
- name: Run UKS tests
5658
run: |
5759
export CUDA_HOME=/usr/local/cuda
@@ -76,6 +78,7 @@ jobs:
7678
pip3 install cupy-cuda12x --upgrade
7779
pip3 install gpu4pyscf-libxc-cuda12x --upgrade
7880
git config --global core.compression 9
81+
python3 -m cupyx.tools.install_library --cuda 12.x --library cutensor
7982
- name: Run TDDFT tests
8083
run: |
8184
export CUDA_HOME=/usr/local/cuda

.github/workflows/unittest.yml

+1
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@ jobs:
5757
pip3 install h5py --upgrade
5858
pip3 install gpu4pyscf-libxc-cuda12x --upgrade
5959
pip3 install cupy-cuda12x --upgrade
60+
python3 -m cupyx.tools.install_library --cuda 12.x --library cutensor
6061
git config --global core.compression 9
6162
- name: Build GPU4PySCF
6263
run: |

builder/build_libxc.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ rm -rf /gpu4pyscf/put4pyscf/lib/*.so
2323

2424
setup_dir=$(dirname $0)
2525

26-
cmake -S /gpu4pyscf/gpu4pyscf/lib -B build/temp.gpu4pyscf-libxc -DBUILD_GINT=OFF -DBUILD_GVHF=OFF -DBUILD_GDFT=OFF -DBUILD_CUPY_HELPER=OFF -DBUILD_SOLVENT=OFF -DBUILD_GVHF_RYS=OFF -DBUILD_GVHF_MD=OFF -DBUILD_PBC=OFF -DCUDA_ARCHITECTURES="70-real"
26+
cmake -S /gpu4pyscf/gpu4pyscf/lib -B build/temp.gpu4pyscf-libxc -DBUILD_GINT=OFF -DBUILD_GVHF=OFF -DBUILD_GDFT=OFF -DBUILD_CUPY_HELPER=OFF -DBUILD_SOLVENT=OFF -DBUILD_GVHF_RYS=OFF -DBUILD_GVHF_MD=OFF -DBUILD_PBC=OFF -DCUDA_ARCHITECTURES="70"
2727
cmake --build build/temp.gpu4pyscf-libxc -j 1
2828

2929
mkdir -p build/lib.gpu4pyscf-libxc/gpu4pyscf/lib/deps/lib

builder/setup_libxc.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535
DOWNLOAD_URL = None
3636
CLASSIFIERS = None
3737
PLATFORMS = None
38-
VERSION = '0.6'
38+
VERSION = '0.7'
3939

4040
def get_cuda_version():
4141
nvcc_out = subprocess.check_output(["nvcc", "--version"]).decode('utf-8')

0 commit comments

Comments
 (0)