torch-mlir

Commit Graph

Author	SHA1	Message	Date
Sambhav Jain	7e0e23c668	Test custom op import with symbolic shapes (#3431 ) Tests the basic constructs of registering a custom op and its abstract implementations (with FakeTensors) in python, going through TorchDynamo export, followed by importing the shape expressions in the Torch dialect. Also fixes the importer were previously the symbolic bind op insertion was not gated in one place.	2024-06-09 00:32:49 -07:00

Author

SHA1

Message

Date

Sambhav Jain

7e0e23c668

Test custom op import with symbolic shapes (#3431 )

Tests the basic constructs of registering a custom op and its abstract
implementations (with FakeTensors) in python, going through TorchDynamo
export, followed by importing the shape expressions in the Torch
dialect.

Also fixes the importer were previously the symbolic bind op insertion
was not gated in one place.

2024-06-09 00:32:49 -07:00

1 Commits (83bfb6fb19c39f8a85ede189483a595944e028ff)