From 71b4f0fae30f248b052e36a96e658517cd6b8ec8 Mon Sep 17 00:00:00 2001 From: "Jiang, Yanbing" Date: Mon, 10 Feb 2025 07:57:11 +0000 Subject: [PATCH] Update (base update) [ghstack-poisoned] --- torch/testing/_internal/common_methods_invocations.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torch/testing/_internal/common_methods_invocations.py b/torch/testing/_internal/common_methods_invocations.py index 68404117327..bd59cdc153d 100644 --- a/torch/testing/_internal/common_methods_invocations.py +++ b/torch/testing/_internal/common_methods_invocations.py @@ -16236,6 +16236,7 @@ op_db: list[OpInfo] = [ DecorateInfo(unittest.skip("Skipped!"), 'TestSchemaCheckModeOpInfo', 'test_schema_correctness'), # aten::_scaled_mm hit the vmap fallback which is currently disabled DecorateInfo(unittest.skip("Skipped!"), "TestVmapOperatorsOpInfo", "test_op_has_batch_rule"), + DecorateInfo(unittest.skip("Skipped!"), "TestVmapOperatorsOpInfo", "test_vmap_exhaustive"), DecorateInfo(unittest.expectedFailure, 'TestNNCOpInfo', 'test_nnc_correctness', dtypes=(torch.float8_e4m3fn, torch.float8_e4m3fnuz, torch.float8_e5m2, torch.float8_e5m2fnuz)), )