Skip to content

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

Merged
chaoming0625 merged 4 commits intomasterfrom
update-taichi-aot-call
Mar 3, 2024
Merged

[math] Support more than 8 parameters of taichi gpu custom operator definition#642
chaoming0625 merged 4 commits intomasterfrom
update-taichi-aot-call

Conversation

@Routhleck
Copy link
Copy Markdown
Member

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 Routhleck marked this pull request as draft March 2, 2024 06:47
@Routhleck Routhleck marked this pull request as ready for review March 3, 2024 05:48
@Routhleck Routhleck requested a review from chaoming0625 March 3, 2024 05:58
@chaoming0625 chaoming0625 merged commit a8377cb into master Mar 3, 2024
@chaoming0625 chaoming0625 deleted the update-taichi-aot-call branch March 3, 2024 06:04
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants