Skip to content

Commit

Permalink
Update operator_custom_with_cupy.ipynb
Browse files Browse the repository at this point in the history
  • Loading branch information
Routhleck committed Mar 25, 2024
1 parent bdce5fc commit 0560709
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions docs/tutorial_advanced/operator_custom_with_cupy.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -29,8 +29,8 @@
"metadata": {},
"source": [
"Although we can now use the flexible taichi custom operator approach, taichi on cuda does not have more fine-grained control or optimization for some scenarios. So for such scenarios, we can use cupy's \n",
"- `RawModule`(https://docs.cupy.dev/en/stable/user_guide/kernel.html#raw-kernels)\n",
"- `jit.rawkernel`(https://docs.cupy.dev/en/stable/user_guide/kernel.html#jit-kernel-definition) \n",
"- [`RawModule`](https://docs.cupy.dev/en/stable/user_guide/kernel.html#raw-kernels)\n",
"- [`jit.rawkernel`](https://docs.cupy.dev/en/stable/user_guide/kernel.html#jit-kernel-definition) \n",
"\n",
"to compile and run CUDA native code directly as strings or cupy JIT function in real time for finer grained control.\n",
"\n",
Expand Down Expand Up @@ -177,8 +177,8 @@
"metadata": {},
"source": [
"尽管我们现在可以使用灵活的taichi自定义操作符方法,但在cuda后端上,taichi没有更细粒度的控制或某些场景下的优化。因此,对于这类场景,我们可以使用cupy的\n",
"- `RawModule`(https://docs.cupy.dev/en/stable/user_guide/kernel.html#raw-kernels)\n",
"- `jit.rawkernel`(https://docs.cupy.dev/en/stable/user_guide/kernel.html#jit-kernel-definition)\n",
"- [`RawModule`](https://docs.cupy.dev/en/stable/user_guide/kernel.html#raw-kernels)\n",
"- [`jit.rawkernel`](https://docs.cupy.dev/en/stable/user_guide/kernel.html#jit-kernel-definition) \n",
"\n",
"来直接作为字符串或cupy JIT函数实时编译并运行CUDA原生代码,以实现更细致的控制。\n",
"\n",
Expand Down

0 comments on commit 0560709

Please sign in to comment.