Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[docstring] compile -> jit in example
Browse files Browse the repository at this point in the history
crcrpar authored Nov 7, 2024
1 parent a0a6151 commit f0c046f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions thunder/core/transforms.py
Original file line number Diff line number Diff line change
@@ -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)

0 comments on commit f0c046f

Please sign in to comment.