You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+5-5Lines changed: 5 additions & 5 deletions
Original file line number
Diff line number
Diff line change
@@ -10,8 +10,8 @@ is intended to be a general purpose model compilation and execution tool.
10
10
Turbine provides three primary tools:
11
11
12
12
**AOT Export*: For compiling one or more `nn.Module`s to compiled, deployment
13
-
ready artifacts. This operates via both a [simple one-shot export API](https://github.com/nod-ai/SHARK-Turbine/blob/main/python/shark_turbine/aot/exporter.py)
14
-
for simple models and an underlying [advanced API](https://github.com/nod-ai/SHARK-Turbine/blob/main/python/shark_turbine/aot/compiled_module.py) for complicated models
13
+
ready artifacts. This operates via both a simple one-shot export API (Already upstreamed to [torch-mlir](https://github.com/llvm/torch-mlir/blob/main/python/torch_mlir/extras/fx_importer.py))
14
+
for simple models and an underlying [advanced API](https://github.com/nod-ai/SHARK-Turbine/blob/main/core/shark_turbine/aot/compiled_module.py) for complicated models
15
15
and accessing the full features of the runtime.
16
16
**Eager Execution*: A `torch.compile` backend is provided and a Turbine Tensor/Device
17
17
is available for more native, interactive use within a PyTorch session.
@@ -62,11 +62,11 @@ compiler, these should be compilable via IREE with `--iree-input-type=torch` for
62
62
end to end execution. Dynamic shape support in torch-mlir is a work in progress,
63
63
and not everything works at head with release binaries at present.
64
64
65
-
*[AOT MLP With Static Shapes](https://github.com/nod-ai/SHARK-Turbine/blob/main/examples/aot_mlp/mlp_export_simple.py)
66
-
*[AOT MLP with a dynamic batch size](https://github.com/nod-ai/SHARK-Turbine/blob/main/examples/aot_mlp/mlp_export_dynamic.py)
65
+
*[AOT MLP With Static Shapes](https://github.com/nod-ai/SHARK-Turbine/blob/main/core/examples/aot_mlp/mlp_export_simple.py)
66
+
*[AOT MLP with a dynamic batch size](https://github.com/nod-ai/SHARK-Turbine/blob/main/core/examples/aot_mlp/mlp_export_dynamic.py)
0 commit comments