CUDA and AMDGPU as weak deps

ComputableDAGs / ComputableDAGs.jl

Representation of Computations as Graphs

https://computabledags.github.io/ComputableDAGs.jl/dev

MIT License

2 stars 2 forks source link

CUDA and AMDGPU as weak deps #2

Closed szabo137 closed 2 months ago

szabo137 commented 2 months ago

Currently, CUDA and AMDGPU are hard dependencies of the package. This makes instantiating of the package really slow, even if one wants to use only the CPU capabilities. I suggest to opt out these dependencies and use PackagesExtensions for the GPU related implementations. Then the user can decide, if CUDA or AMDGPU are needed.