Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Python 3.11 is a strict requirement for building #134

Open
compor opened this issue Oct 3, 2024 · 0 comments
Open

Python 3.11 is a strict requirement for building #134

compor opened this issue Oct 3, 2024 · 0 comments

Comments

@compor
Copy link

compor commented Oct 3, 2024

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant