torch-mlir

Commit Graph

Author	SHA1	Message	Date
Alexandre Rames	d24fa71368	Minor fixes for `ConvertTorchConversionToMLProgram`. (#1991 ) * Only create the global seed variable if it does not exist already. * Make the pass a module pass. A func pass may not modify its parent op.	2023-04-04 09:09:58 -07:00
Vivek Khandelwal	f416953600	[MLIR][TORCH] Add TorchConversionToMLProgram and MLProgramBufferize pass This commit changes the `InsertRngGlobalsPass` to `TorchConversionToMLProgram` pass. This commit also adds the `MLProgramBufferize` pass for the bufferization of ml_program dialect ops to run on refbackend. Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>	2022-12-02 13:20:46 +05:30

Author

SHA1

Message

Date

Alexandre Rames

d24fa71368

Minor fixes for `ConvertTorchConversionToMLProgram`. (#1991 )

* Only create the global seed variable if it does not exist already.
* Make the pass a module pass. A func pass may not modify its parent op.

2023-04-04 09:09:58 -07:00

Vivek Khandelwal

f416953600

[MLIR][TORCH] Add TorchConversionToMLProgram and MLProgramBufferize pass

This commit changes the `InsertRngGlobalsPass` to `TorchConversionToMLProgram`
pass. This commit also adds the `MLProgramBufferize` pass for the
bufferization of ml_program dialect ops to run on refbackend.

Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>

2022-12-02 13:20:46 +05:30

2 Commits (71ca529a62b14135d178fd1c0b643980a233f532)