Skip to content

Commit

Permalink
Merge pull request brainpy#507 from ztqakita/master
Browse files Browse the repository at this point in the history
[docs] add low-level op customization
  • Loading branch information
chaoming0625 authored Sep 21, 2023
2 parents 345c70a + ac93ff7 commit e6373e8
Show file tree
Hide file tree
Showing 3 changed files with 407 additions and 1 deletion.
2 changes: 1 addition & 1 deletion brainpy/_src/math/op_registers/numba_approach/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ class XLACustomOp(BrainPyObject):
cpu_func: callable
The function defines the computation on CPU backend. Same as ``con_compute``.
gpu_func: callable
The function defines the computation on GPU backend. Currently, this function is not supportted.
The function defines the computation on GPU backend. Currently, this function is not supported.
apply_cpu_func_to_gpu: bool
Whether allows to apply CPU function on GPU backend. If True, the GPU data will move to CPU,
and after calculation, the returned outputs on CPU backend will move to GPU.
Expand Down
2 changes: 2 additions & 0 deletions docs/tutorial_advanced/3_dedicated_operators.rst
Original file line number Diff line number Diff line change
Expand Up @@ -3,3 +3,5 @@ Brain Dynamics Dedicated Operators

.. toctree::
:maxdepth: 1

low-level_operator_customization.ipynb
Loading

0 comments on commit e6373e8

Please sign in to comment.