From 213be4b00fb671985da3c004bc94fb956828e1fb Mon Sep 17 00:00:00 2001 From: Chaoming Wang Date: Thu, 18 Jan 2024 23:52:23 +0800 Subject: [PATCH] update --- .../op_register/tests/test_taichi_based.py | 18 ------------------ 1 file changed, 18 deletions(-) diff --git a/brainpy/_src/math/op_register/tests/test_taichi_based.py b/brainpy/_src/math/op_register/tests/test_taichi_based.py index 9ba9b445..2bfe8d31 100644 --- a/brainpy/_src/math/op_register/tests/test_taichi_based.py +++ b/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: