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.

Sign up for GitHub

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

Jump to bottom

[math] Add taichi customized operators(event csrmv, csrmv, jitconn event mv, jitconn mv) #553

Merged
merged 61 commits into from
Dec 29, 2023
Merged

[math] Add taichi customized operators(event csrmv, csrmv, jitconn event mv, jitconn mv) #553

merged 61 commits into from
Dec 29, 2023

Conversation

Routhleck
Copy link
Collaborator

@Routhleck Routhleck commented Dec 1, 2023 •
edited
Loading

@Routhleck Routhleck added the brainpy.math issue belongs to brainpy.math module label Dec 1, 2023
@Routhleck Routhleck self-assigned this Dec 1, 2023
@Routhleck Routhleck marked this pull request as draft December 1, 2023 11:55
Routhleck and others added 16 commits December 2, 2023 15:51
@Routhleck
Fix autograd bug and update test_csrmv_taichi.py
ffccc19
@Routhleck
Fix autograd bug and update test_event_csr_matvec_taichi.py
b05f8c6
@Routhleck
Fix event csr matvec kernel bug
ace8282
@Routhleck
Fix test bugs
6101133
@Routhleck
Add taichi.func random generators
8f3ee1c
@Routhleck
Update test_taichi_random.py
a2518b1
@Routhleck
Implement mv_prob_homo_taichi and mv_prob_uniform_taichi
f3e2171
@Routhleck
Implement jitconn matvec using taichi customized op` and Need to test
08455d4
@Routhleck
Fix bugs in
c0bc79a
@Routhleck
Remove pytest in 'test_taichi_random.py'
6823f34
@Routhleck
Implement jitconn event matvec using taichi customized op and Need to…
8ec4ce8 
… test
@Routhleck
Implement lfsr88 random generator algorithm
81c6722
@Routhleck
Refactor jitconn/_matvec_taichi.py with lfsr88 random generator
e223e69
@chaoming0625
Merge branch 'master' into add-taichi-op
1c39019
@chaoming0625
[csrmv taichi] format codes and redefine JVP rules using .defjvp in…
c3e89db 
…terface
@chaoming0625
[csrmv taichi] format codes of brainpy.math.sparse.csrmv and redefi…
4eb88c1 
…ne JVP rules using `.defjvp` interface
Routhleck and others added 20 commits December 15, 2023 14:12
@Routhleck
Add taichi jitconn matvec benchmark and Optimize taichi jitconn matve…
fd70eea 
…c op
@Routhleck
Refactor taichi event matvec op
074d581
@Routhleck
Add taichi jitconn event matvec benchmark
e9f8b6e
@Routhleck
Optimize taichi jitconn matvec op on CPU backend
66e63d2
@Routhleck
Update taichi jitconn matvec op
2c1a118
@Routhleck
Update test files for taichi jitconn op
e272d15
@Routhleck
Update taichi random generator
1594a2b
@Routhleck
Fix bugs
f3512ce
@Routhleck
Add new function for taichi random seeds initialization
8d90314
@Routhleck
Update taichi_random_time_test.py
5177634
@Routhleck
Update taichi_random_time_test.py
28d8e58
@Routhleck
Update taichi_random_time_test.py
b88b6c2
@Routhleck
Fix bugs
eead6b5
@Routhleck
Remove taichi_random_time_test.py
0829743
@Routhleck
Update test_taichi_random.py
d0788d3
@chaoming0625
[event csr taichi] small upgrade
4bd9bd6
@chaoming0625
[csr mv taichi] fix bugs
c5d55a0
@chaoming0625
[math] new module brainpy.math.tifunc for taichi functionality
ebb7e5b
@chaoming0625
[math] move default environment setting into defaults.py
f840ada
@chaoming0625
[math] fix and update taichi jitconn operators
a14b6a5
@Routhleck Routhleck changed the title [math] Add taichi customized operators [math] Add taichi customized operators(event csrmv, csrmv, jitconn event mv, jitconn mv) Dec 29, 2023
@Routhleck Routhleck mentioned this pull request Dec 29, 2023
Closed
21 tasks
@Routhleck Routhleck marked this pull request as ready for review December 29, 2023 04:28
@Routhleck Routhleck linked an issue Dec 29, 2023 that may be closed by this pull request
[math] Add taichi cutomized operators #569
Closed
21 tasks
@Routhleck Routhleck removed a link to an issue Dec 29, 2023
[math] Add taichi cutomized operators #569
Closed
21 tasks
chaoming0625
chaoming0625 approved these changes Dec 29, 2023
View reviewed changes
Copy link
Collaborator

@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.

So great PR!

@chaoming0625 chaoming0625 merged commit 6368289 into master Dec 29, 2023
32 checks passed
@chaoming0625 chaoming0625 deleted the add-taichi-op branch December 29, 2023 04:33
@chaoming0625 chaoming0625 mentioned this pull request Dec 29, 2023
Closed
2 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@chaoming0625 chaoming0625 chaoming0625 approved these changes

Assignees

@Routhleck Routhleck

Labels
brainpy.math issue belongs to brainpy.math module
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@Routhleck @chaoming0625