torch-mlir

History

Stella Laurenzo 78a3c90758 Add TorchScript graph importer. * Does not handle all features yet but should conservatively fail on unsupported things. * Location tracking is still somewhat mismatched between what TorchScript and MLIR do. Likely need a better heuristic for tracking locations from defs for nodes that do not carry location. * Sets the ground-work for a specialized/generic split but only implements the generic side. * Had some evidence that this requires a recent bump of PT nightly (within the last month) to pick up pybind11 2.6, which includes some cross-module symbol fixes (vs the previously sync'd version). No source changes, but older versions fail to cast function types at runtime.	2020-11-23 14:20:09 -08:00
..
npcomp	Add aten.mm op and "test" it e2e.	2020-11-20 17:21:24 -08:00
npcomp-c	Add TorchScript graph importer.	2020-11-23 14:20:09 -08:00

Stella Laurenzo 78a3c90758 Add TorchScript graph importer.

* Does not handle all features yet but should conservatively fail on unsupported things.
* Location tracking is still somewhat mismatched between what TorchScript and MLIR do. Likely need a better heuristic for tracking locations from defs for nodes that do not carry location.
* Sets the ground-work for a specialized/generic split but only implements the generic side.
* Had some evidence that this requires a recent bump of PT nightly (within the last month) to pick up pybind11 2.6, which includes some cross-module symbol fixes (vs the previously sync'd version). No source changes, but older versions fail to cast function types at runtime.

2020-11-23 14:20:09 -08:00

npcomp

Add aten.mm op and "test" it e2e.

2020-11-20 17:21:24 -08:00

npcomp-c

Add TorchScript graph importer.

2020-11-23 14:20:09 -08:00