This is not fully clear from the listed requirements as stated ATM, that Python 3.11 is the upper version currently supported.
WARNING: Non-homogeneous multireg PERF_COUNTER_ENABLE skip multireg specific data generation.
[15/36] Translating NsNet2 using iree-turbine
FAILED: samples/nsnet2/nsnet2.mlirbc /home/vasich/wb/build-quid2/runtime/samples/nsnet2/nsnet2.mlirbc
cd /home/vasich/wb/build-quid2/runtime/samples/nsnet2 && /home/vasich/wb/quidditch/venv/bin/python3.12 /home/vasich/wb/quidditch/runtime/samples/nsnet2/NsNet2.py /home/vasich/wb/build-quid2/runtime/samples/nsnet2/nsnet2.mlirbc --dtype=f64
Traceback (most recent call last):
File "/home/vasich/wb/quidditch/runtime/samples/nsnet2/NsNet2.py", line 102, in <module>
exported = aot.export(with_frames(n_frames=args.frames))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/vasich/wb/quidditch/venv/lib/python3.12/site-packages/shark_turbine/aot/exporter.py", line 304, in export
cm = TransformedModule(context=context, import_to="import")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/vasich/wb/quidditch/venv/lib/python3.12/site-packages/shark_turbine/aot/compiled_module.py", line 652, in __new__
do_export(proc_def)
File "/home/vasich/wb/quidditch/venv/lib/python3.12/site-packages/shark_turbine/aot/compiled_module.py", line 649, in do_export
trace.trace_py_func(invoke_with_self)
File "/home/vasich/wb/quidditch/venv/lib/python3.12/site-packages/shark_turbine/aot/support/procedural/tracer.py", line 122, in trace_py_func
return_py_value = _unproxy(py_f(*self.proxy_posargs, **self.proxy_kwargs))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/vasich/wb/quidditch/venv/lib/python3.12/site-packages/shark_turbine/aot/compiled_module.py", line 630, in invoke_with_self
return proc_def.callable(self, *args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/vasich/wb/quidditch/runtime/samples/nsnet2/NsNet2.py", line 91, in main
y, out1, out2 = aot.jittable(model.forward)(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/vasich/wb/quidditch/venv/lib/python3.12/site-packages/shark_turbine/aot/support/procedural/base.py", line 135, in __call__
return current_ir_trace().handle_call(self, args, kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/vasich/wb/quidditch/venv/lib/python3.12/site-packages/shark_turbine/aot/support/procedural/tracer.py", line 138, in handle_call
return target.resolve_call(self, *args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/vasich/wb/quidditch/venv/lib/python3.12/site-packages/shark_turbine/aot/builtins/jittable.py", line 228, in resolve_call
gm, guards = exported_f(*flat_pytorch_args)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/vasich/wb/quidditch/venv/lib/python3.12/site-packages/torch/_dynamo/eval_frame.py", line 1202, in inner
check_if_dynamo_supported()
File "/home/vasich/wb/quidditch/venv/lib/python3.12/site-packages/torch/_dynamo/eval_frame.py", line 593, in check_if_dynamo_supported
raise RuntimeError("Python 3.12+ not yet supported for torch.compile")
RuntimeError: Python 3.12+ not yet supported for torch.compile
[17/36] Generating big_matvec/big_matvec_module.h, big_matvec/big_matvec.o, big_matvec/big_matvec.h, big_matvec/big_matvec_llvm.h
ninja: build stopped: subcommand failed.
FAILED: runtime-prefix/src/runtime-stamp/runtime-build /home/vasich/wb/build-quid2/runtime-prefix/src/runtime-stamp/runtime-build
cd /home/vasich/wb/build-quid2/runtime && /usr/bin/cmake --build .
ninja: build stopped: subcommand failed.
It might be worth addressing instead of downgrading Python.
This is not fully clear from the listed requirements as stated ATM, that Python 3.11 is the upper version currently supported.
This hinges on pytorch/pytorch#120233 producing the following error:
It might be worth addressing instead of downgrading Python.