Skip to content

Add CUDA backend support for the heterobatch interface - #92

Open
YuWei-CH wants to merge 8 commits into
flatironinstitute:mainfrom
YuWei-CH:pure-cuda
Open

Add CUDA backend support for the heterobatch interface#92
YuWei-CH wants to merge 8 commits into
flatironinstitute:mainfrom
YuWei-CH:pure-cuda

Conversation

@YuWei-CH

Copy link
Copy Markdown
Contributor

This PR adds native CUDA support for the heterobatch interface.
Specifically, it:

  1. Translates the original CuPy-based implementation into native CUDA, implementing both the finufft_fast and finufft_chi2 interfaces directly in CUDA.
  2. Integrates these CUDA kernels as helper implementations for the heterobatch interface.
  3. Updates CMakeLists.txt to optionally enable the CUDA backend for the heterobatch interface when requested by the user.
  4. Adds documentation for the CUDA backend to README.md, including build and configuration instructions.

Comment thread CMakeLists.txt Outdated
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants