Backport #249
Created by: vchuravy
Backport #249.
@ali-ramadhan
I noticed that you only tested CPU()
execution,
but the changes are limited to CUDAKernels.
Created by: vchuravy
Backport #249.
@ali-ramadhan
I noticed that you only tested CPU()
execution,
but the changes are limited to CUDAKernels.