Skip to content

Commit

Permalink
Avoid executor imports
Browse files Browse the repository at this point in the history
  • Loading branch information
carmocca committed May 6, 2024
1 parent 3e8cc63 commit 059941f
Showing 1 changed file with 2 additions and 7 deletions.
9 changes: 2 additions & 7 deletions extensions/thunder/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -482,12 +482,10 @@ Thunder allows you to define a priority list of executors that can map operators

```python
import thunder
from thunder.executors.sdpaex import sdpa_ex
from thunder.executors.torch_compile import torch_compile_cat_ex

model = thunder.jit(
model,
executors=[sdpa_ex, torch_compile_cat_ex, thunder.nvfuser_executor, thunder.pytorch_executor]
executors=["sdpa", "torchcompile_cat", "nvfuser", "torch"]
)
```

Expand All @@ -508,13 +506,10 @@ We can enable this executor by passing it to the list of executors available. Th

```python
import thunder
from thunder.executors.sdpaex import sdpa_ex
from thunder.executors.torch_compile import torch_compile_cat_ex
from unsloth.executor import unsloth_ex

model = thunder.jit(
model,
executors=[sdpa_ex, unsloth_ex, torch_compile_cat_ex, thunder.nvfuser_executor, thunder.pytorch_executor]
executors=["sdpa", "unsloth", "torchcompile_cat", "nvfuser", "torch"]
)
```

Expand Down

0 comments on commit 059941f

Please sign in to comment.