Skip to content

Commit

Permalink
Update operator_custom_with_taichi.ipynb
Browse files Browse the repository at this point in the history
  • Loading branch information
Routhleck committed May 15, 2024
1 parent e61db3f commit 3820aa7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/tutorial_advanced/operator_custom_with_taichi.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -127,7 +127,7 @@
"metadata": {},
"source": [
"### Registering and Using Custom Operators\n",
"After defining a custom operator, it can be registered into a specific framework and used where needed. When registering, you can specify cpu_kernel and gpu_kernel, so the operator can run on different devices. Specify the outs parameter when calling, using jax.ShapeDtypeStruct to define the shape and data type of the output.\n",
"After defining a custom operator, it can be registered into a specific framework and used where needed. When registering, you can specify cpu_kernel and gpu_kernel, so the operator can run on different devices. Specify the outs parameter when calling, using `jax.ShapeDtypeStruct` to define the shape and data type of the output.\n",
"\n",
"Note: Maintain the order of the operator's declared parameters consistent with the order when calling.\n",
"\n",
Expand Down

0 comments on commit 3820aa7

Please sign in to comment.