Skip to content

Python 3.11 is a strict requirement for building #134

@compor

Description

@compor

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:

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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions