You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Having multiple distribution type for inputs fails gpu kernels due to Union split limiation. Currently, can have at most 3 types in the Union of a CuArray for cuda kernels. JuliaGPU/CUDA.jl#1385.
khosravipasha
changed the title
Having multiple distribution type for inputs fails gpu kernels due to Union split limiation
Allow Multiple InputNode type for CUDA Kernels
Mar 17, 2022
Having multiple distribution type for inputs fails gpu kernels due to Union split limiation. Currently, can have at most 3 types in the Union of a CuArray for cuda kernels. JuliaGPU/CUDA.jl#1385.
Another example here
For
nodes
, we already haveSum
andMul
node types, so we can at most have one InputNode type for our cuda kernels.Possible Workarounds:
The text was updated successfully, but these errors were encountered: