Why functions in SpecialFunctions package work on CUDA arrays?

singularity
Fri, 18 Apr 2025 22:54:55 +0000
Performance

HI. I am confused how the behavior of functions inside the SpecialFunctions package on CUDA arrays. I am confused why the code below works and doesn’t throw an error at the last line.

using CUDA
using SpecialFunctions
x = rand(Float32, 64, 64, 25)
x_gpu = cu(x)
z = loggamma.(x_gpu)

8 posts - 5 participants