torch-mlir/test/python
Ramiro Leal-Cavazos e0f301c890
Add `extra_library` kwarg to `torch_mlir.compile` (#1986)
This commit adds the ability to specify extra abstract interpretation
functions in `torch_mlir.compile` to use during type refinement. This
allows users to easily add custom ops without having to interact with
MLIR or C++ directly.
2023-03-30 09:20:19 -07:00
..
importer/jit_ir Fix Base Lazy Backend Type Conversion (#1412) 2022-10-04 15:53:28 -07:00
custom_op_shape_dtype_fn.py Add `extra_library` kwarg to `torch_mlir.compile` (#1986) 2023-03-30 09:20:19 -07:00
lit.local.cfg Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
smoketest.py WIP: No PyTorch dep (#1854) 2023-02-13 14:21:06 -06:00