From 6484776a25b6cb181e7083efd6795a74a5ca186c Mon Sep 17 00:00:00 2001 From: Sean Silva Date: Wed, 3 Aug 2022 17:32:28 +0000 Subject: [PATCH] Make numerical stability test more perverse To test the summation stability of `torch.aten.var`, add a large constant to it, which increases the effective precision requirements. --- python/torch_mlir_e2e_test/test_suite/stats.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/torch_mlir_e2e_test/test_suite/stats.py b/python/torch_mlir_e2e_test/test_suite/stats.py index 3e7afb381..1093a90aa 100644 --- a/python/torch_mlir_e2e_test/test_suite/stats.py +++ b/python/torch_mlir_e2e_test/test_suite/stats.py @@ -688,4 +688,4 @@ class VarCorrectionLargeInputModule(torch.nn.Module): @register_test_case(module_factory=lambda: VarCorrectionLargeInputModule()) def VarCorrectionLargeInputModule_basic(module, tu: TestUtils): - module.forward(tu.rand(3, 4, 1024, 8192)) + module.forward(100 + tu.rand(3, 4, 1024, 8192))