Is it possible to use CuStaticSharedArray(T, n) with n const?

Specific Domains GPU
,
2

Hi,

Yes, this is certainly possible in the style of

const B = 16

function kernel_test!(P)
    v_shared = CuStaticSharedArray(Float64, (B + 2, B + 2, 2))
    ...
    return
end

If you want to experiment with different values of B, it might make sense to (temporarily) use something like

function kernel_test!(P, ::Val{B}) where B
    v_shared = CuStaticSharedArray(Float64, (B + 2, B + 2, 2))
    ...
    return
end

...
@cuda ... kernel_test!(P, Val(16))
1 Like