| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
[TORCH] Modified fx_importer to support hop_while_loop (#4338) This PR adds support for emitting graphs for Pytorch HOPs, beginning with torch._higher_order_ops.while_loop. The proposed change is to modify the import_program to call function _import_all_child_modules, which recursively imports the stateless graph for all the children modules. Since HOP operator graphs are stateless graphs with no mutation, it is correct to import them as stateless graphs, although the method import_stateless_graph is marked as "deprecated". --------- Signed-off-by: Keshav Vinayak Jha <keshavvinayakjha@gmail.com> | 8 个月前 | |
Include torch-mlir-opt in Python wheels (#3964) This adds the torch-mlir-opt tool to the Python wheels, which allows to use the commandline tool via a pip installed package instead of having to compile the torch-mlir project yourself. The executable is still installed to the deault location and copied over via the setup.py to be included in the Python wheel. This could be refactored and handled within CMake in a follow-up. | 1 年前 | |
[FxImporter] Fix sympy_int_to_int utility (#3657) New sympy type is introduced to represent integer infinity in upstream PyTorch repo. Subsequently, sympy.oo is no longer used to represent infinity upper bound for dynamic dimensions where the upper bound is unknown. Instead int_oo is used to represent integer infinity. This commit updates the _sympy_int_to_int utility in light of this change. | 1 年前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 8 个月前 | ||
| 1 年前 | ||
| 1 年前 |