Skip to content

Commit 20819c4

Browse files
committed
[bug] fix
1 parent 4b0d0f3 commit 20819c4

File tree

5 files changed

+450
-453
lines changed

5 files changed

+450
-453
lines changed

brainpy/_src/layers/linear.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -207,3 +207,21 @@ def __init__(self, *args, **kwargs) -> None:
207207

208208
def update(self, x):
209209
return x
210+
211+
212+
class CSRLinear(Layer):
213+
pass
214+
215+
216+
class CSCLinear(Layer):
217+
pass
218+
219+
220+
class BSRLinear(Layer):
221+
pass
222+
223+
224+
class MatLinear(Layer):
225+
pass
226+
227+

brainpy/_src/math/event/tests/test_csrmv_gpu.py renamed to brainpy/_src/math/event/tests/test_event_csrmv_gpu.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,12 +4,12 @@
44
import jax
55
import pytest
66

7-
import test_csrmv
7+
import test_event_csrmv
88

99
if jax.default_backend() != 'gpu':
1010
pytest.skip("No gpu available.", allow_module_level=True)
1111

1212

13-
class Test_event_csr_matvec_GPU(test_csrmv.Test_event_csr_matvec):
13+
class Test_event_csr_matvec_GPU(test_event_csrmv.Test_event_csr_matvec):
1414
def __init__(self, *args, **kwargs):
1515
super().__init__(*args, **kwargs, platform='gpu')

0 commit comments

Comments
 (0)