Skip to content

Commit d401506

Browse files
committed
Update test names
1 parent 125bfe9 commit d401506

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
@@ -2848,7 +2848,7 @@
28482848
"AvgPool1dPadCeilPadNotIncluded_basic",
28492849
"AvgPool2dDiffKernelsStridesPadCeilPadNotIncluded_basic",
28502850
"AvgPool3dDiffKernelsStridesPadCeilPadNotIncluded_basic",
2851-
"AvgPool3dSingleIntTupleParamsModule_basic",
2851+
"AvgPool3dSingleIntTupleStrideModule_basic",
28522852
"BatchMlpLayerModule_basic",
28532853
"BincountMinlengthModule_basic",
28542854
"BincountModule_basic",
@@ -3040,7 +3040,7 @@
30403040
"MaxPool2dWithIndicesNonDefaultDilationModule_basic",
30413041
"MaxPool2dWithIndicesNonDefaultParamsModule_basic",
30423042
"MaxPool2dWithIndicesNonDefaultStrideModule_basic",
3043-
"MaxPool2dSingleIntTupleParamsModule_basic",
3043+
"MaxPool2dSingleIntTupleKernelModule_basic",
30443044
"MaxPool3dCeilModeTrueModule_basic",
30453045
"MaxPool3dLargeDatadModule_basic",
30463046
"MaxPool3dModuleRandomSimple_basic",
@@ -3052,7 +3052,7 @@
30523052
"MaxPool3dWithIndicesNonDefaultDilationModule_basic",
30533053
"MaxPool3dWithIndicesNonDefaultParamsModule_basic",
30543054
"MaxPool3dWithIndicesNonDefaultStrideModule_basic",
3055-
"MaxPool3dSingleIntTupleParamsModule_basic",
3055+
"MaxPool3dSingleIntTupleDilationModule_basic",
30563056
"MaxUnpool3dModule_basic",
30573057
"MaxUnpool3dModulePad0_basic",
30583058
"MaxUnpool2dModule_basic",
@@ -3553,7 +3553,7 @@
35533553
"AvgPool3dStaticModule_basic",
35543554
"AvgPool3dCountIncludePadFalse_basic",
35553555
"AvgPool3dCountIncludePadFalseWithoutPadding_basic",
3556-
"AvgPool3dSingleIntTupleParamsModule_basic",
3556+
"AvgPool3dSingleIntTupleStrideModule_basic",
35573557
"Conv_Transpose1dModule_basic",
35583558
"Conv_Transpose1dStaticModule_basic",
35593559
"Conv_Transpose2dStaticModule_basic",
@@ -3809,7 +3809,7 @@
38093809
"MaxPool3dWithIndicesNonDefaultParamsModule_basic",
38103810
"MaxPool3dWithIndicesNonDefaultStrideModule_basic",
38113811
"MaxPool3dWithIndicesStaticModule_basic",
3812-
"MaxPool3dSingleIntTupleParamsModule_basic",
3812+
"MaxPool3dSingleIntTupleDilationModule_basic",
38133813
"MeanDimEmptyDimModule_basic",
38143814
"MlGroupNormManualModule_basic",
38153815
"MlGroupNormModule_basic",
@@ -4253,7 +4253,7 @@
42534253
"AvgPool2dIntModule_basic",
42544254
"AvgPool2dStaticModule_basic",
42554255
"AvgPool2dWithoutPadModule_basic",
4256-
"AvgPool3dSingleIntTupleParamsModule_basic",
4256+
"AvgPool3dSingleIntTupleStrideModule_basic",
42574257
"BatchMlpLayerModule_basic",
42584258
"BernoulliFloatModule_basic",
42594259
"BernoulliModule_basic",
@@ -4661,7 +4661,7 @@
46614661
"MaxPool3dWithIndicesNonDefaultParamsModule_basic",
46624662
"MaxPool3dWithIndicesNonDefaultStrideModule_basic",
46634663
"MaxPool3dWithIndicesStaticModule_basic",
4664-
"MaxPool3dSingleIntTupleParamsModule_basic",
4664+
"MaxPool3dSingleIntTupleDilationModule_basic",
46654665
"MeanDimAllReduceKeepdimModule_basic",
46664666
"MeanDimAllReduceModule_basic",
46674667
"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
@@ -550,7 +550,7 @@ def MaxPool2dCeilModeFullDimIndivisibleByStrideModule_basic(module, tu: TestUtil
550550
module.forward(tu.rand(1, 1, 75, 75, low=-1))
551551

552552

553-
class MaxPool2dSingleIntTupleParamsModule(torch.nn.Module):
553+
class MaxPool2dSingleIntTupleKernelModule(torch.nn.Module):
554554
def __init__(self):
555555
super().__init__()
556556
self.mpd = torch.nn.MaxPool2d(
@@ -570,8 +570,8 @@ def forward(self, x):
570570
return self.mpd(x)
571571

572572

573-
@register_test_case(module_factory=lambda: MaxPool2dSingleIntTupleParamsModule())
574-
def MaxPool2dSingleIntTupleParamsModule_basic(module, tu: TestUtils):
573+
@register_test_case(module_factory=lambda: MaxPool2dSingleIntTupleKernelModule())
574+
def MaxPool2dSingleIntTupleKernelModule_basic(module, tu: TestUtils):
575575
module.forward(tu.rand(2, 4, 20, 20, low=0.5, high=1.0))
576576

577577

@@ -747,7 +747,7 @@ def MaxPool3dCeilModeTrueModule_basic(module, tu: TestUtils):
747747
module.forward(tu.rand(1, 1, 20, 20, 20, low=0.5, high=1.0))
748748

749749

750-
class MaxPool3dSingleIntTupleParamsModule(torch.nn.Module):
750+
class MaxPool3dSingleIntTupleDilationModule(torch.nn.Module):
751751
def __init__(self):
752752
super().__init__()
753753
self.mpd = torch.nn.MaxPool3d(
@@ -768,8 +768,8 @@ def forward(self, x):
768768
return self.mpd(x)
769769

770770

771-
@register_test_case(module_factory=lambda: MaxPool3dSingleIntTupleParamsModule())
772-
def MaxPool3dSingleIntTupleParamsModule_basic(module, tu: TestUtils):
771+
@register_test_case(module_factory=lambda: MaxPool3dSingleIntTupleDilationModule())
772+
def MaxPool3dSingleIntTupleDilationModule_basic(module, tu: TestUtils):
773773
module.forward(tu.rand(2, 4, 20, 20, 20, low=0.5, high=1.0))
774774

775775

@@ -1861,7 +1861,7 @@ def AvgPool3dCountIncludePadFalseWithoutPadding_basic(module, tu: TestUtils):
18611861
module.forward(tu.rand(3, 3, 12, 12, 12, low=-1))
18621862

18631863

1864-
class AvgPool3dSingleIntTupleParamsModule(torch.nn.Module):
1864+
class AvgPool3dSingleIntTupleStrideModule(torch.nn.Module):
18651865
def __init__(self):
18661866
super().__init__()
18671867
self.apd = torch.nn.AvgPool3d(
@@ -1881,8 +1881,8 @@ def forward(self, x):
18811881
return self.apd(x)
18821882

18831883

1884-
@register_test_case(module_factory=lambda: AvgPool3dSingleIntTupleParamsModule())
1885-
def AvgPool3dSingleIntTupleParamsModule_basic(module, tu: TestUtils):
1884+
@register_test_case(module_factory=lambda: AvgPool3dSingleIntTupleStrideModule())
1885+
def AvgPool3dSingleIntTupleStrideModule_basic(module, tu: TestUtils):
18861886
module.forward(tu.rand(2, 4, 20, 20, 20, low=0.5, high=1.0))
18871887

18881888

0 commit comments

Comments
 (0)