Open
Description
Calculating the exponential of a GPU matrix yields errors:
using CUDA
A = cu([1.0, 1;0 1])
CUDA.exp(A)
results in an error:
ERROR: ArgumentError: cannot take the CPU address of a CuArray{Float32, 2, CUDA. Mem.DeviceBuffer}
Trying to do this via ExponentialUtils
also throws errors.
It seems that the function called does not work properly for GPU arrays, either because indexing is done explicitly somewhere, or because some precompiled library is called that can only handle CPU arrays. This seems like a basic functionality, so I'm not sure if I'm requesting a new feature, if this is a bug, or if I'm just doing something wrong.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment