Code

Cuda zero copy#13002

Code

Closed

scottrbrtsn wants to merge 28 commits intomne-tools:mainmne-tools/mne-python:mainfrom scottrbrtsn:cuda-zero-copyscottrbrtsn/mne-python:cuda-zero-copyCopy head branch name to clipboard

+45-4

Commits

Commits on Dec 6, 2024

• update mne/cuda to call cupy.asarray when possible

  scottrbrtsn

  committed
  c6cd611

• update tests as poc, they pass

  scottrbrtsn

  committed
  02edb9c

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 5662420
  

  pre-commit-ci[bot]

  authored and

  scottrbrtsn

  committed
  5662420

• Update mne/cuda.py

  

  larsoner

  authored and

  scottrbrtsn

  committed
  f7a04ad

• string formatting

  scottrbrtsn

  committed
  fd0463e

• add changelog file

  scottrbrtsn

  committed
  a03014e

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 44b8b9a
  

  pre-commit-ci[bot]

  authored and

  scottrbrtsn

  committed
  44b8b9a

• only get shared mem if cuda capable

  scottrbrtsn

  committed
  96f2a03

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for caec586
  

  pre-commit-ci[bot]

  authored and

  scottrbrtsn

  committed
  caec586

• revert numba_capable and use the bool in fixes

  scottrbrtsn

  committed
  de298fe

• gate get_shared_mem with fixes/has_numba bool

  scottrbrtsn

  committed
  ce2ff37

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 75c4dab
  

  pre-commit-ci[bot]

  authored and

  scottrbrtsn

  committed
  75c4dab

• Update mne/cuda.py to soft import numba

  Show description for 929fcb9
  

  scottrbrtsn

  and

  larsoner

  committed
  929fcb9

• revert tests, make shared_mem fun private and simplified, call appropriately within the cuda module.

  scottrbrtsn

  committed
  214edf1

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for cfd6fb3
  

  pre-commit-ci[bot]

  authored and

  scottrbrtsn

  committed
  cfd6fb3

• soft import numba

  scottrbrtsn

  committed
  cb8fec5

• remove extra line

  scottrbrtsn

  committed
  6973b54

• revert soft import.

  scottrbrtsn

  committed
  f2950e0

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for ac5b5c8
  

  pre-commit-ci[bot]

  authored and

  scottrbrtsn

  committed
  ac5b5c8

• W needs "cuda" passed in since it just set n_jobs to 1

  scottrbrtsn

  committed
  e76b5a7

• hfft needs cuda for the same reason

  scottrbrtsn

  committed
  1fc59ba

• remove n_jobs param

  scottrbrtsn

  committed
  9350498

• remove n_jobs gate

  scottrbrtsn

  committed
  1a1c8ec

• fix docstring param

  scottrbrtsn

  committed
  a3342fa

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 88ae8cb
  

  pre-commit-ci[bot]

  authored and

  scottrbrtsn

  committed
  88ae8cb

• irfft, takes cuda mem, and so x is a different type. i do not think getting shared mem is necessary, for filtering...need to think on this tho

  scottrbrtsn

  committed
  4bbd2c7

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for af3cab2
  pre-commit-ci[bot]

  committed
  af3cab2

• remove troubleshooting woes

  scottrbrtsn

  committed
  74a7794