| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
fix: remove split.Tensor and split_with_sizes from decomp tables (#4340) More details are in the issue: https://github.com/llvm/torch-mlir/issues/4339 Since split.Tensor and split_with_sizes decompose to as_strided (see issue above why as_strided is problematic), we remove them from the decomp tables. cc @zjgarvey --------- Signed-off-by: raayandhar <rdhar@amd.com> | 9 个月前 | |
[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 个月前 | |
python/torch_mlir/extras/onnx_importer.py: fix: SyntaxWarning: invalid escape sequence '\w' (#4296) Mark raw string to avoid syntax warning Co-authored-by: Florian Walbroel <walbroel@roofline.ai> | 10 个月前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 9 个月前 | ||
| 9 个月前 | ||
| 10 个月前 |