Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Merged
merged 4 commits into from
Mar 3, 2024

Conversation

Routhleck
Copy link
Collaborator

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
34 of 37 checks passed
@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