From 1577a5bb6275b3154d03ec1ab9ce021c7788c3a9 Mon Sep 17 00:00:00 2001 From: Garrett Wright Date: Fri, 12 Jan 2024 10:53:47 -0500 Subject: [PATCH] adds CUDA 12 extension support, changes GHA workflow to new gpu_12x extension --- .github/workflows/workflow.yml | 2 +- pyproject.toml | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/workflow.yml b/.github/workflows/workflow.yml index 44d799e4ae..4a70116185 100644 --- a/.github/workflows/workflow.yml +++ b/.github/workflows/workflow.yml @@ -128,7 +128,7 @@ jobs: - uses: actions/checkout@v3 - name: Install dependencies run: | - pip install -e ".[dev,gpu_11x]" + pip install -e ".[dev,gpu_12x]" - name: Customize config run: | echo "Setup tmp dirs and chmod so others can cleanup." diff --git a/pyproject.toml b/pyproject.toml index 8af01f30ed..71baee3982 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -67,6 +67,7 @@ gpu_102 = ["pycuda", "cupy-cuda102", "cufinufft==1.3"] gpu_110 = ["pycuda", "cupy-cuda110", "cufinufft==1.3"] gpu_111 = ["pycuda", "cupy-cuda111", "cufinufft==1.3"] gpu_11x = ["pycuda", "cupy-cuda11x", "cufinufft==1.3"] +gpu_12x = ["pycuda", "cupy-cuda12x", "cufinufft==1.3"] dev = [ "black", "bumpversion",