JuliaGPU / DaggerGPU.jl

GPU integrations for Dagger.jl
Other
50 stars 11 forks source link

Add Datadeps support #36

Closed jpsamaroo closed 4 months ago

jpsamaroo commented 4 months ago

Primarily adds Dagger.move! methods, but in the process, also switches to a single-stream-per-device model for simplicity. Also adds new functionality to Dagger.scope such that Dagger.scope(cuda_gpus=:) or Dagger.scope(cuda_gpus=[1,3]) works.