mirror of https://github.com/llvm/torch-mlir
[cleanup] Remove disabled e2e test
This test has been disabled a long time, and since RefBackend is so slow we don't want to add this unnecessarily. I believe it is covered by downstream testing such as the Shark Tank.pull/1696/head
parent
d52359a891
commit
c956c39c86
|
@ -84,28 +84,6 @@ class IouOfModule(torch.nn.Module):
|
||||||
def IouOfModule_basic(module, tu: TestUtils):
|
def IouOfModule_basic(module, tu: TestUtils):
|
||||||
module.forward(tu.rand(1024, 4), tu.rand(1024, 4))
|
module.forward(tu.rand(1024, 4), tu.rand(1024, 4))
|
||||||
|
|
||||||
class MobilenetV2Module(torch.nn.Module):
|
|
||||||
def __init__(self):
|
|
||||||
super().__init__()
|
|
||||||
# Reset seed to make model deterministic.
|
|
||||||
torch.manual_seed(0)
|
|
||||||
self.mobilenetv2 = models.mobilenet_v2()
|
|
||||||
self.train(False)
|
|
||||||
|
|
||||||
@export
|
|
||||||
@annotate_args([
|
|
||||||
None,
|
|
||||||
([-1, 3, -1, -1], torch.float32, True),
|
|
||||||
])
|
|
||||||
def forward(self, img):
|
|
||||||
return self.mobilenetv2.forward(img)
|
|
||||||
|
|
||||||
# TODO (cathyzhyi) The runtime assertion for conv2d with group != 1 is exposed
|
|
||||||
# after aten.hardtanh is implemented. Reenable once the the runtime assertion
|
|
||||||
# is fixed.
|
|
||||||
#@register_test_case(module_factory=lambda: MobilenetV2Module())
|
|
||||||
def MobilenetV2Module_basic(module, tu: TestUtils):
|
|
||||||
module.forward(tu.rand(1, 3, 224, 224))
|
|
||||||
|
|
||||||
class MobilenetV3Module(torch.nn.Module):
|
class MobilenetV3Module(torch.nn.Module):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
|
|
Loading…
Reference in New Issue