Skip to content

Conversation

@stellaraccident
Copy link
Collaborator

@stellaraccident stellaraccident commented Nov 19, 2023

This lifts the core of the jit_ir_importer and ltc out of the pt1 project, making them peers to it. As a side-effect of this layering, now the "MLIR bits" (dialects, etc) are not commingled with the various parts of the pt1 project, allowing pt1 and ltc to overlay cleanly onto a more fundamental "just MLIR" Python core. Prior to this, the Python namespace was polluted to the point that this could not happen.

That "just MLIR" Python core will be introduced in a followup, which will create the space to upstream the FX and ONNX pure Python importers.

This primary non-NFC change to the API is:

  • torch_mlir.dialects.torch.importer.jit_ir -> torch_mlir.jit_ir_importer.

The rest is source code layering so that we can make the pt1 project optional without losing the other features.

Progress on #2546.

@stellaraccident stellaraccident force-pushed the breakup_python_pytorch_deps branch from b6e7e13 to 1fcaf74 Compare November 19, 2023 01:10
@stellaraccident stellaraccident force-pushed the breakup_python_pytorch_deps branch from 1fcaf74 to e9a9b62 Compare November 19, 2023 01:25
@stellaraccident stellaraccident marked this pull request as ready for review November 19, 2023 04:07
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

Successfully merging this pull request may close these issues.

3 participants