BrainPy
BrainPy copied to clipboard
[math] Support more than 8 parameters of taichi gpu custom operator definition
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):
...