| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
[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> | 9 个月前 | |
[ONNX] Fix import of boolean tensor constants (#4064) Currently the shape and element type of boolean tensor attributes are incorrect, since they use the shape/dtype of the packed numpy array rather than of the original tensor value. Specifying the shape and element type explicitly resolves this. | 1 年前 | |
Add Windows build workflow in the CI. (#4041) 1. Adds a workflow to trigger windows build in the CI similar to the existing one for Linux to address the build CI request captured in https://github.com/llvm/torch-mlir/issues/3985 2. Only non-python based LIT unit tests are enabled in the workflow due to failures reported in https://github.com/llvm/torch-mlir/issues/4040 3. The CI is only run against torch-nightly since python based tests are disabled. My understanding is that without the python based tests there's no need to run against the stable version too. If that's not the case, will enable stable as well. Tested that both nightly and stable versions work fine: https://github.com/sahas3/torch-mlir/actions/runs/13477124004 | 1 年前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 9 个月前 | ||
| 1 年前 | ||
| 1 年前 |