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

[Stablehlo] lowering aten.log1p to stablehlo.log_plus_one (#3110)

pull/3113/head
Yuanqiang Liu 2024-04-07 17:01:58 +08:00 committed by GitHub
parent 0a00f38a7e
commit 498ab997cd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lib/Conversion/TorchToStablehlo/Basic.cpp
View File

@ -1784,6 +1784,7 @@ void mlir::torch::torch_to_stablehlo::populateBasicOpPatternsAndLegality(
patterns.add<ConvertAtenUnaryFPOnlyOp<AtenOp, StablehloOp>>(typeConverter, \
context)
INSERT_UNARY_FPONLY_PATTERN(AtenLogOp, stablehlo::LogOp);
INSERT_UNARY_FPONLY_PATTERN(AtenLog1pOp, stablehlo::Log1pOp);
INSERT_UNARY_FPONLY_PATTERN(AtenExpOp, stablehlo::ExpOp);
INSERT_UNARY_FPONLY_PATTERN(AtenSqrtOp, stablehlo::SqrtOp);
INSERT_UNARY_FPONLY_PATTERN(AtenRsqrtOp, stablehlo::RsqrtOp);

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

@ -887,6 +887,7 @@ STABLEHLO_PASS_SET = {
"ViewOffsetTestStaticModule_basic",
"ViewTwoFiveThreeStaticModule_basic",
"ViewTwoToThreeStaticModule_basic",
"ElementwiseLog1pModule_basic",
}
STABLEHLO_CRASHING_SET = {