From db84e15eaace0aaa6b55385d5389352d5a35a38f Mon Sep 17 00:00:00 2001 From: Masaki Kozuki Date: Mon, 11 Nov 2024 17:10:33 +0900 Subject: [PATCH] [docstring] `compile` -> `jit` in example (#1409) --- thunder/core/transforms.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/thunder/core/transforms.py b/thunder/core/transforms.py index d59fad498b..2bf91372fe 100644 --- a/thunder/core/transforms.py +++ b/thunder/core/transforms.py @@ -2969,12 +2969,12 @@ def forward_and_backward_from_trace(trace: Trace, torch_autograd=False) -> Forwa Example: >>> import torch - >>> from thunder import compile, last_traces + >>> from thunder import jit, last_traces >>> from thunder.core.transforms import forward_and_backward_from_trace >>> def f(x): ... return torch.sin(x) >>> x = torch.tensor(3.0) - >>> cf = compile(f) + >>> cf = jit(f) >>> out = cf(x) >>> trace = last_traces(cf)[0] >>> forward_and_backward_from_trace(trace)