| commit | 2f47c081bc9506dd583a8249e06745b83dde79b1 | [log] [tgz] |
|---|---|---|
| author | Scott Todd <scotttodd@google.com> | Thu Nov 09 11:03:36 2023 -0800 |
| committer | GitHub <noreply@github.com> | Thu Nov 09 11:03:36 2023 -0800 |
| tree | 0e5ad021eab34145dc06c12192c670eebfa7de73 | |
| parent | 094af3dea2584acbc69c84c318aacf9cf295b070 [diff] [blame] |
Add pytorch_aot_advanced sample Colab notebook using SHARK-Turbine. (#15329) Progress on https://github.com/openxla/iree/issues/15117 This is similar to pytorch_aot_simple but uses a few features from the advanced (`aot.CompiledModule`) API. * This is a pretty minimal example for completeness and an easy branching off point. Examples like https://github.com/nod-ai/SHARK-Turbine/blob/main/examples/llama2_inference/stateless_llama.py are better references for how to use these APIs with a real program. Preview URL for review: https://colab.research.google.com/github/scotttodd/iree/blob/pytorch-samples-3/samples/colab/pytorch_aot_advanced.ipynb skip-ci: not covered by presubmit
diff --git a/samples/colab/test_notebooks.py b/samples/colab/test_notebooks.py index 9918fcd..97decb0 100755 --- a/samples/colab/test_notebooks.py +++ b/samples/colab/test_notebooks.py
@@ -20,6 +20,7 @@ # samples.Dockerfile does not currently include. "tflite_text_classification.ipynb", # PyTorch notebooks using SHARK-Turbine require Python 3.10+ in Docker. + "pytorch_aot_advanced.ipynb", "pytorch_aot_simple.ipynb", "pytorch_jit.ipynb", ]