[math] Refactor taichi operators

brainpy / BrainPy

Brain Dynamics Programming in Python

https://brainpy.readthedocs.io/

GNU General Public License v3.0

515 stars 93 forks source link

Closed Routhleck closed 8 months ago

Routhleck commented 8 months ago

Add dnn.linear taichi implmentation

Routhleck commented 8 months ago

Replace with taichi op

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

Routhleck commented 8 months ago

Set all operators to using taichi kernel defaultly