torch-mlir

Commit Graph

Author	SHA1	Message	Date
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

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

1 Commits (cc06391630dc6d2f189787389873a6310212fba6)