From fae0d3357996f08943f7dc5ee7ff2fb412c26e49 Mon Sep 17 00:00:00 2001 From: Dan F-M Date: Thu, 14 Mar 2024 14:12:33 -0400 Subject: [PATCH] includes --- lib/jax_finufft_gpu.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/lib/jax_finufft_gpu.cc b/lib/jax_finufft_gpu.cc index 072db48..147db4e 100644 --- a/lib/jax_finufft_gpu.cc +++ b/lib/jax_finufft_gpu.cc @@ -2,6 +2,7 @@ // It is exposed as a standard pybind11 module defining "capsule" objects containing our // method. For simplicity, we export a separate capsule for each supported dtype. +#include "cufinufft.h" #include "kernels.h" #include "pybind11_kernel_helpers.h"