llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[FxImporter] Fix failed e2e case (#3365)

pull/3334/head
penguin_wwy 2024-05-22 00:20:54 +08:00 committed by GitHub
parent b870729efe
commit c2c1c2cfa4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
projects/pt1/e2e_testing/xfail_sets.py
View File

@ -455,9 +455,6 @@ FX_IMPORTER_XFAIL_SET = {
"ThresholdBackward2dMixedModule_basic",
"TorchPrimLoopForLikeModule_basic",
"TorchPrimLoopWhileLikeModule_basic",
"UnbindIntGetItem_Module_basic",
"UnbindIntListUnpack_Module_basic",
"UnsafeIndexPutHackedTwin1DFloatNonAccumulateModule_basic",
"UnsafeViewCollapseDynamicWithAtenSizeIntModule_basic",
"UpSampleNearest2dDynamicFactor_basic",
"ViewCollapseDynamicWithAtenSizeIntModule_basic",

1
python/torch_mlir/extras/fx_decomp_util.py
View File

@ -47,6 +47,7 @@ DEFAULT_DECOMPOSITIONS = [
torch.ops.aten.linspace.default,
torch.ops.aten.triu.default,
torch.ops.aten.nan_to_num.default,
torch.ops.aten.unbind,
]

2
python/torch_mlir/extras/fx_importer.py
View File

@ -1428,6 +1428,8 @@ class GraphNodeImporter:
elif target == torch.ops.aten._assert_async.msg:
# TODO: A more suitable op to replace it?
return
elif target == torch.ops.aten._unsafe_index_put.default:
node.target = target = torch.ops.aten._unsafe_index_put.hacked_twin
schema = target._schema
assert isinstance(schema, FunctionSchema)