torch-mlir/python/test/compile_api
Sean Silva 6403c0e56f torch_mlir.compile: allow custom backend_legal_ops set
Allow customizing `backend_legal_ops` for "torch" output type, since we
don't know which backend will be used (it might be a custom backend).
We don't allow customizing the `backend_legal_ops` for the other output
types (Linalg, TOSA, MHLO) since those backends control their set of
legal ops directly.

Fixes #1418
2022-10-12 04:21:22 -07:00
..
backend_legal_ops.py torch_mlir.compile: allow custom backend_legal_ops set 2022-10-12 04:21:22 -07:00
basic.py Update test to pass with newer versions of tanh (#990) 2022-06-28 20:28:13 -07:00
output_type_spec.py torch_mlir.compile: Allow OutputType as a string. 2022-07-08 17:37:27 -07:00
tracing.py Reject dictionary inputs when tracing. 2022-09-30 16:02:35 -04:00