Skip to content

[math] Refactor taichi operators#598

Merged
chaoming0625 merged 27 commits intomasterfrom
add-dnn-linear-taichi
Jan 29, 2024
Merged

[math] Refactor taichi operators#598
chaoming0625 merged 27 commits intomasterfrom
add-dnn-linear-taichi

Conversation

@Routhleck
Copy link
Copy Markdown
Member

Add dnn.linear taichi implmentation

Copy link
Copy Markdown
Member Author

@Routhleck Routhleck left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Ready to pull request

@Routhleck Routhleck marked this pull request as ready for review January 25, 2024 09:35
Copy link
Copy Markdown
Member

@chaoming0625 chaoming0625 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I am confused by the code in this PR. The discussion may be needed.

@Routhleck Routhleck changed the title [dnn] Add dnn.linear taichi implmentation [dnn] Refactor taichi operators Jan 27, 2024
@Routhleck Routhleck marked this pull request as draft January 27, 2024 06:00
@Routhleck
Copy link
Copy Markdown
Member Author

Routhleck commented Jan 28, 2024

Replace with taichi op

  • event csr matvec
    cpu default: taichi
    gpu default: brainpylib(cusparse)
  • csr matvec
    cpu default: taichi
    gpu default: taichi
  • jitconn event matvec
    cpu default: taichi
    gpu default: taichi(outdim parallel=False) brainpylib(outdim parallel=True)
  • jitconn matvec
    cpu default: taichi
    gpu default: taichi(outdim parallel=False) brainpylib(outdim parallel=True)

@Routhleck Routhleck marked this pull request as ready for review January 28, 2024 15:09
@Routhleck Routhleck changed the title [dnn] Refactor taichi operators [math] Refactor taichi operators Jan 28, 2024
@Routhleck
Copy link
Copy Markdown
Member Author

Set all operators to using taichi kernel defaultly

Copy link
Copy Markdown
Member

@chaoming0625 chaoming0625 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Great!

@chaoming0625 chaoming0625 merged commit 7e8dd81 into master Jan 29, 2024
@chaoming0625 chaoming0625 deleted the add-dnn-linear-taichi branch January 29, 2024 15:15
@Routhleck Routhleck added enhancement New feature or request brainpy.math issue belongs to brainpy.math module labels Feb 3, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

brainpy.math issue belongs to brainpy.math module enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants