From 55b43e3d22fdb4796f9d7a5af3a5d7c07f5740cc Mon Sep 17 00:00:00 2001 From: "cq_chu@163.com" Date: Thu, 22 Feb 2024 17:41:35 +0800 Subject: [PATCH] docs(xla): test ci --- source/user-guide/model-development/jit/xla.rst | 1 + 1 file changed, 1 insertion(+) diff --git a/source/user-guide/model-development/jit/xla.rst b/source/user-guide/model-development/jit/xla.rst index e56afc94d5a..afc37768c0c 100644 --- a/source/user-guide/model-development/jit/xla.rst +++ b/source/user-guide/model-development/jit/xla.rst @@ -123,6 +123,7 @@ XLA executable 加速训练过程。 模型训练迭代中存在动态执行逻辑的情况下， 无法将整个计算交由 XLA 执行。 这种情况下可以使用 jit.patrial_trace 装饰器对其中静态的部分进行加速。 + 被 partial_trace 包装部分的前向/反向会使用 XLA 执行, 其他部分仍由 MegEngine 执行。 代码示例如下：