Skip to content

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

Merged
chaoming0625 merged 61 commits intomasterfrom
add-taichi-op
Dec 29, 2023
Merged

[math] Add taichi customized operators(event csrmv, csrmv, jitconn event mv, jitconn mv)#553
chaoming0625 merged 61 commits intomasterfrom
add-taichi-op

Conversation

@Routhleck
Copy link
Member

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

So great PR!

@chaoming0625 chaoming0625 merged commit 6368289 into master Dec 29, 2023
@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