Skip to content
GPU Build

Updating CMake infrastructure to use upstream CMakeLists #4

Sign in to view logs

Updating CMake infrastructure to use upstream CMakeLists

Updating CMake infrastructure to use upstream CMakeLists #4

Workflow file for this run

.github/workflows/gpu-build.yml at 7faa3f3
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: Get the version info for NVCC
run: nvcc --version
- uses: actions/checkout@v4
with:
submodules: true
fetch-depth: 0
- uses: actions/setup-python@v4
with:
python-version: 3.11
- name: Install Python dependencies
run: python -m pip install -U pip
- name: Compile the extension
run: python -m pip install -v .
- name: Check that the GPU extension was built
run: python -c "import jax_finufft.jax_finufft_gpu"