Skip to content

Bump pypa/gh-action-pypi-publish from 1.8.14 to 1.9.0 #199

Bump pypa/gh-action-pypi-publish from 1.8.14 to 1.9.0

Bump pypa/gh-action-pypi-publish from 1.8.14 to 1.9.0 #199

Workflow file for this run

name: GPU Build
on:
push:
branches:
- main
pull_request:
jobs:
build:
runs-on: ubuntu-22.04
container: nvidia/cuda:12.2.2-devel-ubuntu22.04
steps:
- name: Install git and FFTW
run: apt-get update && apt-get install -y git libfftw3-dev
env:
DEBIAN_FRONTEND: "noninteractive"
- name: Get the version info for NVCC
run: nvcc --version
- uses: actions/checkout@v4
with:
submodules: true
fetch-depth: 0
- uses: actions/setup-python@v5
with:
python-version: 3.11
- name: Install Python dependencies
run: |
python -m pip install -U pip
python -m pip install -U jax[cpu]
- name: Compile the extension
run: python -m pip install -v .
env:
CMAKE_ARGS: "-DJAX_FINUFFT_USE_CUDA=ON"
- name: Check that the GPU extension was built
run: python -c "import jax_finufft.jax_finufft_gpu"