diff --git a/lib/Conversion/TorchToLinalg/Random.cpp b/lib/Conversion/TorchToLinalg/Random.cpp index aa4ec91d7da5..854e3f86d367 100644 --- a/lib/Conversion/TorchToLinalg/Random.cpp +++ b/lib/Conversion/TorchToLinalg/Random.cpp @@ -186,7 +186,7 @@ class ConvertAtenUniformOp : public OpConversionPattern { Value res = randomUniformF64(b, loc, linearIndex, key, min, max); Value truncRes = res; - if (isa(elemTy)) + if (isa(elemTy)) truncRes = b.create(loc, elemTy, res); b.create(loc, truncRes); }) diff --git a/projects/pt1/e2e_testing/xfail_sets.py b/projects/pt1/e2e_testing/xfail_sets.py index 4abe6e48ee88..e818dbc6ae2b 100644 --- a/projects/pt1/e2e_testing/xfail_sets.py +++ b/projects/pt1/e2e_testing/xfail_sets.py @@ -33,6 +33,8 @@ # if a dimension is specified in all expand lists, and not in sumdim list. # This is a bug in the implementation of _trilinear in PyTorch. "Aten_TrilinearModuleZerodDimBug_basic", + # missing lowering from aten.pow.Tensor_Tensor for integer result + "PowIntIntModule_basic", } if torch_version_for_comparison() < version.parse("2.5.0.dev"): @@ -220,7 +222,6 @@ "AtenIntBoolOpConstFalseModule_basic", "AtenIntBoolOpConstTrueModule_basic", "IntFloatModule_basic", - "PowIntFloatModule_basic", # END tests failing due to: torch._dynamo.exc.Unsupported: torch.* op returned non-Tensor int call_function aten.Int # ERROR: torch._dynamo.exc.Unsupported: torch.* op returned non-Tensor int call_function aten.len "LenStrModule_basic", @@ -857,7 +858,6 @@ "NormalFunctionalModule_basic", "NumelModule_basic", "NumelZeroRankModule_basic", - "PowIntFloatModule_basic", "PrimMaxIntModule_basic", "PrimMinIntDynamicModule_basic", "PrimMinIntModule_basic", @@ -2373,6 +2373,8 @@ "PadWithNoneValModule_basic", "PermuteModule_basic", "PermuteNegativeIndexModule_basic", + "PowFloatFloatModule_basic", + "PowFloatIntModule_basic", "PrimListUnpackNumMismatchModule_basic", "PrimsIotaModule_basic", "PrimsSqueezeEmptyDimensionsModule_basic", @@ -3153,6 +3155,7 @@ "PixelShuffleModuleSpatiallyDynamic_basic", "PixelShuffleModuleSpatiallyStatic_basic", "PixelShuffleModuleStaticRank3Int64_basic", + "PowIntIntModule_basic", "PrimMaxIntModule_basic", "PrimMinIntDynamicModule_basic", "PrimMinIntModule_basic", @@ -4777,7 +4780,6 @@ "PixelShuffleModuleSpatiallyStatic_basic", "PixelShuffleModuleStaticRank3Int64_basic", "PixelShuffleModuleStaticRank4Float32_basic", - "PowIntFloatModule_basic", "PrimMaxIntModule_basic", "PrimMinIntDynamicModule_basic", "PrimMinIntModule_basic",