Skip to content

Commit

Permalink
docs(xla): test ci
Browse files Browse the repository at this point in the history
  • Loading branch information
cqchu committed Feb 22, 2024
1 parent d2a2d0f commit 55b43e3
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions source/user-guide/model-development/jit/xla.rst
Original file line number Diff line number Diff line change
Expand Up @@ -123,6 +123,7 @@ XLA executable 加速训练过程。

模型训练迭代中存在动态执行逻辑的情况下, 无法将整个计算交由 XLA 执行。
这种情况下可以使用 jit.patrial_trace 装饰器对其中静态的部分进行加速。

被 partial_trace 包装部分的前向/反向会使用 XLA 执行, 其他部分仍由 MegEngine 执行。
代码示例如下:

Expand Down

0 comments on commit 55b43e3

Please sign in to comment.