torch-mlir

Commit Graph

Author	SHA1	Message	Date
Stella Laurenzo	6877302504	[NFC reformat] Applies pre-commit formatting to Python files. (#3244 ) This is a large change because prior to this point, Python files in the project were not consistently formatted. This reformats them all with black defaults. Based on experience with prior projects, if you have a dev/long-term branch with Python patches, you can minimize merge conflicts prior to rebasing to include this commit by running `black` on your modified Python files, squashing, and then rebasing/merging.	2024-04-27 14:16:31 -07:00
saienduri	bfcf93ea21	Rename torch_mlir.compile APIs and introduce FX based analogs (#2842 ) Link to related RFC: https://discourse.llvm.org/t/rfc-rename-torch-mlir-compile-apis-and-introduce-fx-based-analogs/76646 This commit updates the documentation, tests, CMake files, and API for the proposed changes in the RFC. There is a new torch_mlir/fx.py for user level APIs related to importing modules and a corresponding test for this path can be found at test/python/fx_importer/basic_test.py. --------- Co-authored-by: MaheshRavishankar <mravisha@amd.com>	2024-02-06 19:07:59 -08:00
Rik Huijzer	8328998172	Allow printing all IR in `torch_mlir.compile` (#2669 ) This PR adds the `enable_ir_printing` option to `torch_mlir.compile`, which can be used to print the IR for all intermediate passes. When running the added test file via: ```shell $ python test/python/compile.py 2> tiny.stderr ``` the file `tiny.stderr` is about 700 KB.	2023-12-20 15:08:21 -06:00

Author

SHA1

Message

Date

Stella Laurenzo

6877302504

[NFC reformat] Applies pre-commit formatting to Python files. (#3244 )

This is a large change because prior to this point, Python files in the
project were not consistently formatted. This reformats them all with
black defaults.

Based on experience with prior projects, if you have a dev/long-term
branch with Python patches, you can minimize merge conflicts prior to
rebasing to include this commit by running `black` on your modified
Python files, squashing, and then rebasing/merging.

2024-04-27 14:16:31 -07:00

saienduri

bfcf93ea21

Rename torch_mlir.compile APIs and introduce FX based analogs (#2842 )

Link to related RFC:
https://discourse.llvm.org/t/rfc-rename-torch-mlir-compile-apis-and-introduce-fx-based-analogs/76646
This commit updates the documentation, tests, CMake files, and API for
the proposed changes in the RFC. There is a new torch_mlir/fx.py for
user level APIs related to importing modules and a corresponding test
for this path can be found at test/python/fx_importer/basic_test.py.

---------

Co-authored-by: MaheshRavishankar <mravisha@amd.com>

2024-02-06 19:07:59 -08:00

Rik Huijzer

8328998172

Allow printing all IR in `torch_mlir.compile` (#2669 )

This PR adds the `enable_ir_printing` option to `torch_mlir.compile`,
which can be used to print the IR for all intermediate passes.

When running the added test file via:
```shell
$ python test/python/compile.py 2> tiny.stderr
```
the file `tiny.stderr` is about 700 KB.

2023-12-20 15:08:21 -06:00

3 Commits (0abc5868b5f5cdc0abd123f6dc03853c6bd8c39d)