Code

add cuda backend for KLU#35

Code

Open

joamatab wants to merge 9 commits intoflaport:mainflaport/sax:mainfrom joamatab:cudajoamatab/sax:cudaCopy head branch name to clipboard

+449-3

Commits

Commits on May 22, 2024

• badd cuda

  joamatab

  committed
  80cf2cb

  Copy full SHA for 80cf2cb

• add cuda to optional packages

  joamatab

  committed
  734c32b

  Copy full SHA for 734c32b

• Add CUDA backend.

  bdice

  committed
  be4acb9

  Copy full SHA for be4acb9

• Merge branch 'cuda' of github.com:joamatab/sax into cuda

  bdice

  committed
  bf2d1a7

  Copy full SHA for bf2d1a7

• Use jax[cuda12].

  bdice

  committed
  b1955a8

  Copy full SHA for b1955a8

• Add backends test script.

  bdice

  committed
  6e6c693

  Copy full SHA for 6e6c693

• Apply pre-commit to test_nbs.py.

  bdice

  committed
  e3743b8

  Copy full SHA for e3743b8

Commits on May 23, 2024

• Use loops to handle missing dimension in CUDA backend.

  bdice

  committed
  5078cd6

  Copy full SHA for 5078cd6

• Add quick start test.

  bdice

  committed
  d7ba9c2

  Copy full SHA for d7ba9c2