update

brainpy/_src/math/op_register/tests/test_taichi_based.py

@@ -1,29 +1,11 @@
 import jax
 import jax.numpy as jnp
 import taichi as taichi
-import pytest
-import platform
 
 import brainpy.math as bm
 
 bm.set_platform('cpu')
 
-# if not platform.platform().startswith('Windows'):
-#   pytest.skip(allow_module_level=True)
-
-
-# @ti.kernel
-# def event_ell_cpu(indices: ti.types.ndarray(ndim=2),
-#                   vector: ti.types.ndarray(ndim=1),
-#                   weight: ti.types.ndarray(ndim=1),
-#                   out: ti.types.ndarray(ndim=1)):
-#   weight_0 = weight[0]
-#   num_rows, num_cols = indices.shape
-#   ti.loop_config(serialize=True)
-#   for i in range(num_rows):
-#     if vector[i]:
-#       for j in range(num_cols):
-#         out[indices[i, j]] += weight_0
 
 @taichi.func
 def get_weight(weight: taichi.types.ndarray(ndim=1)) -> taichi.f32: