Skip to content

Commit 3e003bb

Browse files
committed
Update test names
1 parent b667200 commit 3e003bb

File tree

2 files changed

+16
-16
lines changed

2 files changed

+16
-16
lines changed

projects/pt1/e2e_testing/xfail_sets.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2859,7 +2859,7 @@
28592859
"AvgPool1dPadCeilPadNotIncluded_basic",
28602860
"AvgPool2dDiffKernelsStridesPadCeilPadNotIncluded_basic",
28612861
"AvgPool3dDiffKernelsStridesPadCeilPadNotIncluded_basic",
2862-
"AvgPool3dSingleIntTupleParamsModule_basic",
2862+
"AvgPool3dSingleIntTupleStrideModule_basic",
28632863
"BatchMlpLayerModule_basic",
28642864
"BincountMinlengthModule_basic",
28652865
"BincountModule_basic",
@@ -3056,7 +3056,7 @@
30563056
"MaxPool2dWithIndicesNonDefaultDilationModule_basic",
30573057
"MaxPool2dWithIndicesNonDefaultParamsModule_basic",
30583058
"MaxPool2dWithIndicesNonDefaultStrideModule_basic",
3059-
"MaxPool2dSingleIntTupleParamsModule_basic",
3059+
"MaxPool2dSingleIntTupleKernelModule_basic",
30603060
"MaxPool3dCeilModeTrueModule_basic",
30613061
"MaxPool3dLargeDatadModule_basic",
30623062
"MaxPool3dModuleRandomSimple_basic",
@@ -3068,7 +3068,7 @@
30683068
"MaxPool3dWithIndicesNonDefaultDilationModule_basic",
30693069
"MaxPool3dWithIndicesNonDefaultParamsModule_basic",
30703070
"MaxPool3dWithIndicesNonDefaultStrideModule_basic",
3071-
"MaxPool3dSingleIntTupleParamsModule_basic",
3071+
"MaxPool3dSingleIntTupleDilationModule_basic",
30723072
"MaxUnpool3dModule_basic",
30733073
"MaxUnpool3dModulePad0_basic",
30743074
"MaxUnpool2dModule_basic",
@@ -3578,7 +3578,7 @@
35783578
"AvgPool3dStaticModule_basic",
35793579
"AvgPool3dCountIncludePadFalse_basic",
35803580
"AvgPool3dCountIncludePadFalseWithoutPadding_basic",
3581-
"AvgPool3dSingleIntTupleParamsModule_basic",
3581+
"AvgPool3dSingleIntTupleStrideModule_basic",
35823582
"Conv_Transpose1dModule_basic",
35833583
"Conv_Transpose1dStaticModule_basic",
35843584
"Conv_Transpose3dModule_basic",
@@ -3821,7 +3821,7 @@
38213821
"MaxPool3dWithIndicesNonDefaultParamsModule_basic",
38223822
"MaxPool3dWithIndicesNonDefaultStrideModule_basic",
38233823
"MaxPool3dWithIndicesStaticModule_basic",
3824-
"MaxPool3dSingleIntTupleParamsModule_basic",
3824+
"MaxPool3dSingleIntTupleDilationModule_basic",
38253825
"MeanDimEmptyDimModule_basic",
38263826
"MlGroupNormManualModule_basic",
38273827
"MlGroupNormModule_basic",
@@ -4271,7 +4271,7 @@
42714271
"AvgPool2dIntModule_basic",
42724272
"AvgPool2dStaticModule_basic",
42734273
"AvgPool2dWithoutPadModule_basic",
4274-
"AvgPool3dSingleIntTupleParamsModule_basic",
4274+
"AvgPool3dSingleIntTupleStrideModule_basic",
42754275
"BatchMlpLayerModule_basic",
42764276
"BernoulliFloatModule_basic",
42774277
"BernoulliModule_basic",
@@ -4686,7 +4686,7 @@
46864686
"MaxPool3dWithIndicesNonDefaultParamsModule_basic",
46874687
"MaxPool3dWithIndicesNonDefaultStrideModule_basic",
46884688
"MaxPool3dWithIndicesStaticModule_basic",
4689-
"MaxPool3dSingleIntTupleParamsModule_basic",
4689+
"MaxPool3dSingleIntTupleDilationModule_basic",
46904690
"MeanDimAllReduceKeepdimModule_basic",
46914691
"MeanDimAllReduceModule_basic",
46924692
"MeanDimDtypeModule_basic",

projects/pt1/python/torch_mlir_e2e_test/test_suite/pooling.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -599,7 +599,7 @@ def MaxPool2dCeilModeFullDimIndivisibleByStrideModule_basic(module, tu: TestUtil
599599
module.forward(tu.rand(1, 1, 75, 75, low=-1))
600600

601601

602-
class MaxPool2dSingleIntTupleParamsModule(torch.nn.Module):
602+
class MaxPool2dSingleIntTupleKernelModule(torch.nn.Module):
603603
def __init__(self):
604604
super().__init__()
605605
self.mpd = torch.nn.MaxPool2d(
@@ -619,8 +619,8 @@ def forward(self, x):
619619
return self.mpd(x)
620620

621621

622-
@register_test_case(module_factory=lambda: MaxPool2dSingleIntTupleParamsModule())
623-
def MaxPool2dSingleIntTupleParamsModule_basic(module, tu: TestUtils):
622+
@register_test_case(module_factory=lambda: MaxPool2dSingleIntTupleKernelModule())
623+
def MaxPool2dSingleIntTupleKernelModule_basic(module, tu: TestUtils):
624624
module.forward(tu.rand(2, 4, 20, 20, low=0.5, high=1.0))
625625

626626

@@ -796,7 +796,7 @@ def MaxPool3dCeilModeTrueModule_basic(module, tu: TestUtils):
796796
module.forward(tu.rand(1, 1, 20, 20, 20, low=0.5, high=1.0))
797797

798798

799-
class MaxPool3dSingleIntTupleParamsModule(torch.nn.Module):
799+
class MaxPool3dSingleIntTupleDilationModule(torch.nn.Module):
800800
def __init__(self):
801801
super().__init__()
802802
self.mpd = torch.nn.MaxPool3d(
@@ -817,8 +817,8 @@ def forward(self, x):
817817
return self.mpd(x)
818818

819819

820-
@register_test_case(module_factory=lambda: MaxPool3dSingleIntTupleParamsModule())
821-
def MaxPool3dSingleIntTupleParamsModule_basic(module, tu: TestUtils):
820+
@register_test_case(module_factory=lambda: MaxPool3dSingleIntTupleDilationModule())
821+
def MaxPool3dSingleIntTupleDilationModule_basic(module, tu: TestUtils):
822822
module.forward(tu.rand(2, 4, 20, 20, 20, low=0.5, high=1.0))
823823

824824

@@ -1910,7 +1910,7 @@ def AvgPool3dCountIncludePadFalseWithoutPadding_basic(module, tu: TestUtils):
19101910
module.forward(tu.rand(3, 3, 12, 12, 12, low=-1))
19111911

19121912

1913-
class AvgPool3dSingleIntTupleParamsModule(torch.nn.Module):
1913+
class AvgPool3dSingleIntTupleStrideModule(torch.nn.Module):
19141914
def __init__(self):
19151915
super().__init__()
19161916
self.apd = torch.nn.AvgPool3d(
@@ -1930,8 +1930,8 @@ def forward(self, x):
19301930
return self.apd(x)
19311931

19321932

1933-
@register_test_case(module_factory=lambda: AvgPool3dSingleIntTupleParamsModule())
1934-
def AvgPool3dSingleIntTupleParamsModule_basic(module, tu: TestUtils):
1933+
@register_test_case(module_factory=lambda: AvgPool3dSingleIntTupleStrideModule())
1934+
def AvgPool3dSingleIntTupleStrideModule_basic(module, tu: TestUtils):
19351935
module.forward(tu.rand(2, 4, 20, 20, 20, low=0.5, high=1.0))
19361936

19371937

0 commit comments

Comments
 (0)