Passing device array to routines
Open
- Dominant language
- Python
- Stars
- 99
- Forks
- 11
- PR merge metrics
- No merged PRs in 30d
Description
Hello!
I would like to know if it is possible to pass arrays that are on the device to the routines. For example something like this (that I tried but is failing),
A_d = cuda.to_device(A)
B_d = cuda.to_device(A)
C_h = cublas.gemm("N", "N", 1.0, A, B)
C_h = cuda.copy_to_host(C_d)
This could be useful in order to avoid memory transfer if the data need to be use again later.
Best regards,
Marc
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.