Skip to content

Allow Multiple InputNode type for CUDA Kernels #120

Open
@khosravipasha

Description

@khosravipasha

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 have Sum and Mul node types, so we can at most have one InputNode type for our cuda kernels.

Possible Workarounds:

  • Merge all input node types into one
  • Merge Sum and Mul nodes into one, (then we can have two input types at most)

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions