From f36388ece2ab4bcadac05a6f59d14e0357b8d759 Mon Sep 17 00:00:00 2001 From: Matthias Gehre Date: Wed, 11 Dec 2024 16:02:29 +0100 Subject: [PATCH] Use existing test --- projects/pt1/e2e_testing/xfail_sets.py | 2 +- .../torch_mlir_e2e_test/test_suite/basic.py | 24 ------------------- 2 files changed, 1 insertion(+), 25 deletions(-) diff --git a/projects/pt1/e2e_testing/xfail_sets.py b/projects/pt1/e2e_testing/xfail_sets.py index 3f988e4700de..3606e48d8996 100644 --- a/projects/pt1/e2e_testing/xfail_sets.py +++ b/projects/pt1/e2e_testing/xfail_sets.py @@ -1993,6 +1993,7 @@ "ElementwisePowTensorBroadcastModule_basic", "ElementwisePowTensorBroadcastStaticModule_basic", "ElementwisePowTensorModule_basic", + "ElementwisePowTensorStaticModule_basic", "ElementwisePreluModule_basic", "ElementwisePreluStaticModule_basic", "ElementwiseRad2DegModule_basic", @@ -2161,7 +2162,6 @@ "PermuteNegativeIndexModule_basic", "PowFloatFloatModule_basic", "PowFloatIntModule_basic", - "PowBroadcastModule_basic", "PrimListUnpackNumMismatchModule_basic", "PrimsIotaModule_basic", "PrimsSqueezeEmptyDimensionsModule_basic", diff --git a/projects/pt1/python/torch_mlir_e2e_test/test_suite/basic.py b/projects/pt1/python/torch_mlir_e2e_test/test_suite/basic.py index 97f2b9457674..a46ac5a571c5 100644 --- a/projects/pt1/python/torch_mlir_e2e_test/test_suite/basic.py +++ b/projects/pt1/python/torch_mlir_e2e_test/test_suite/basic.py @@ -4551,30 +4551,6 @@ def PowFloatFloatModule_basic(module, tu: TestUtils): # ============================================================================== -class PowBroadcastModule(torch.nn.Module): - def __init__(self): - super().__init__() - - @export - @annotate_args( - [ - None, - ([-1, -1, -1], torch.float32, True), - ([], torch.float32, True), - ] - ) - def forward(self, x, y): - return torch.ops.aten.pow(x, y) - - -@register_test_case(module_factory=lambda: PowBroadcastModule()) -def PowBroadcastModule_basic(module, tu: TestUtils): - module.forward(tu.rand(3, 4, 5), torch.ones([])) - - -# ============================================================================== - - class PowIntFloatModule(torch.nn.Module): def __init__(self): super().__init__()