update for CUDA 3.3 (@workspace -> with_workspace)

Jutho / TensorOperations.jl

Julia package for tensor contractions and related operations

https://jutho.github.io/TensorOperations.jl/stable/

Other

443 stars 56 forks source link

update for CUDA 3.3 (@workspace -> with_workspace) #109

Closed jfeist closed 2 years ago

jfeist commented 2 years ago

This uses CUDA.with_workspace if it exists (CUDA 3.3 and later), and defines a minimal wrapper that uses CUDA.@workspace if not. I checked that tests pass on CUDA 2.6, 3.2, 3.3, and 3.4.

Fixes #107

Jutho commented 2 years ago

Thanks for tackling this. Looks great.

alhirzel commented 2 years ago

@Jutho - any chance this PR could be incorporated into a release of TensorOperations.jl? Thank you for your work, and making these tools so smooth and easy to use!