Skip to content

Commit 5481ec4

Browse files
committed
fix op register bug
1 parent 9f6040b commit 5481ec4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

brainpy/math/operators/op_register.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ def register_op(
134134
_check_brainpylib(register_op.__name__)
135135
f = brainpylib.register_op(name,
136136
cpu_func=cpu_func,
137-
gpu_func=gpu_func,
137+
gpu_func_translation=gpu_func,
138138
out_shapes=eval_shape,
139139
apply_cpu_func_to_gpu=apply_cpu_func_to_gpu)
140140

0 commit comments

Comments
 (0)