mirror of https://github.com/llvm/torch-mlir
78a3c90758
* 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. |
||
|---|---|---|
| .. | ||
| test_arange.py | ||
| test_conv_nllloss_grads.py | ||
| test_export_ResA.py | ||
| test_export_add3.py | ||
| test_export_batchnorm.py | ||
| test_export_cat.py | ||
| test_export_conv2d_fwd.py | ||
| test_export_multi_out.py | ||