BrainPy icon indicating copy to clipboard operation
BrainPy copied to clipboard

[math] Support more than 8 parameters of taichi gpu custom operator definition

Open Routhleck opened this issue 1 year ago • 0 comments

Support more than 8 parameters of taichi gpu custom operator definition. (Need brainpylib > 0.2.8) The taichi.kernel which register in BrainPy on cuda backend now support more than 8 parameters.

import taichi as ti
@ti.kernel
def foo(param1: ti.types.ndarray(ndim=1),
        param2: ti.types.ndarray(ndim=1),
        param3: ti.types.ndarray(ndim=1),
        param4: ti.types.ndarray(ndim=1),
        param5: ti.types.ndarray(ndim=1),
        param6: ti.types.ndarray(ndim=1),
        param7: ti.types.ndarray(ndim=1),
        param8: ti.types.ndarray(ndim=1),
        param9: ti.types.ndarray(ndim=1),
        ...,
        param32: ti.types.ndarray(ndim=1) # The most parameters supported by the LLVM-based backend): 
  ...

Routhleck avatar Mar 02 '24 06:03 Routhleck