Code

Basic CUDA support#68

Code

Merged

WardBrian merged 17 commits intomainflatironinstitute/pytorch-finufft:mainfrom feat/cuda-startflatironinstitute/pytorch-finufft:feat/cuda-startCopy head branch name to clipboardOct 6, 2023

+364-125

Commits

Commits on Oct 6, 2023

• MVP: cuda support for type 1

  Show description for b50ddcd
  WardBrian

  committed
  b50ddcd

  Copy full SHA for b50ddcd

• Add (failing) tests

  WardBrian

  committed
  7ca3c6f

  Copy full SHA for 7ca3c6f

• First pass at CI

  WardBrian

  committed
  c8c9e0c

  Copy full SHA for c8c9e0c

• Work around modeord issue, cuda forward tests passing

  WardBrian

  committed
  a89b8ef

  Copy full SHA for a89b8ef

• Skip cuda tests in GHA

  WardBrian

  committed
  6937124

  Copy full SHA for 6937124

• MVP cuda backward

  WardBrian

  committed
  6bd21c2

  Copy full SHA for 6bd21c2

• Minor clean up

  WardBrian

  committed
  fd1f729

  Copy full SHA for fd1f729

• Factor out common test code

  WardBrian

  committed
  b66a421

  Copy full SHA for b66a421

• Formatting

  WardBrian

  committed
  052cf41

  Copy full SHA for 052cf41

• Consolidate yum installs

  WardBrian

  committed
  8d8f748

  Copy full SHA for 8d8f748

• CI: Jenkinsfile work

  WardBrian

  committed
  ed5c4e5

  Copy full SHA for ed5c4e5

• CI: Fix email

  WardBrian

  committed
  5492243

  Copy full SHA for 5492243

• Lint fixes

  WardBrian

  committed
  41cab36

  Copy full SHA for 41cab36

• Try k40s

  WardBrian

  authored
  2a4f17d

  Copy full SHA for 2a4f17d

• Back to v100

  WardBrian

  authored
  80a0dcc

  Copy full SHA for 80a0dcc

• CI: Build finufft as well as cufinufft

  WardBrian

  authored
  03c79d6

  Copy full SHA for 03c79d6

• CI: tweaks

  WardBrian

  committed
  f431920

  Copy full SHA for f431920