Skip to content

Commit

Permalink
Use existing test
Browse files Browse the repository at this point in the history
  • Loading branch information
mgehre-amd committed Dec 11, 2024
1 parent 8a224ba commit f36388e
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 25 deletions.
2 changes: 1 addition & 1 deletion projects/pt1/e2e_testing/xfail_sets.py
Original file line number Diff line number Diff line change
Expand Up @@ -1993,6 +1993,7 @@
"ElementwisePowTensorBroadcastModule_basic",
"ElementwisePowTensorBroadcastStaticModule_basic",
"ElementwisePowTensorModule_basic",
"ElementwisePowTensorStaticModule_basic",
"ElementwisePreluModule_basic",
"ElementwisePreluStaticModule_basic",
"ElementwiseRad2DegModule_basic",
Expand Down Expand Up @@ -2161,7 +2162,6 @@
"PermuteNegativeIndexModule_basic",
"PowFloatFloatModule_basic",
"PowFloatIntModule_basic",
"PowBroadcastModule_basic",
"PrimListUnpackNumMismatchModule_basic",
"PrimsIotaModule_basic",
"PrimsSqueezeEmptyDimensionsModule_basic",
Expand Down
24 changes: 0 additions & 24 deletions projects/pt1/python/torch_mlir_e2e_test/test_suite/basic.py
Original file line number Diff line number Diff line change
Expand Up @@ -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__()
Expand Down

0 comments on commit f36388e

Please sign in to comment.