Skip to content

Commit 670c33c

Browse files
authored
[samples] Update PyTorch JIT notebook for Python 3.12 (iree-org#22209)
The PyTorch JIT sample notebook fails on Python 3.12 with `RuntimeError: Dynamo is not supported on Python 3.12+` because it installs PyTorch 2.3.0, which lacks Python 3.12 support. Additionally, the expected MLIR output shown in the notebook does not appear when running with newer versions. Update the PyTorch installation to use the latest stable version from the CPU index, which supports Python 3.12. Add the `TURBINE_LOG_LEVEL=debug` environment variable to enable verbose compilation output, including intermediate MLIR representations. Update the sample execution outputs to reflect the new versions. Signed-off-by: HeatCrab <[email protected]>
1 parent 1ea209d commit 670c33c

File tree

1 file changed

+192
-176
lines changed

1 file changed

+192
-176
lines changed

0 commit comments

Comments
 (0)