skip le 1.10

alibaba · Jul 25, 2023 · ad15c05 · ad15c05
1 parent 1f8957a
commit ad15c05
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pytorch_blade/tests/disc/ops/test_input_mutation.py b/pytorch_blade/tests/disc/ops/test_input_mutation.py
@@ -34,6 +34,7 @@ def tearDown(self):
         del os.environ["TORCH_MHLO_OP_WHITE_LIST"]
         del os.environ["TORCH_BLADE_EXPERIMENTAL_MERGE_HORIZONTAL_GROUPS"]
 
+    @skipTorchLE("1.10.0")
     def test_inplace_kv(self):
         k_cache = torch.zeros(2, 32, 8, device=self.device)
         k = torch.ones(2, 1, 8, device=self.device)
@@ -46,5 +47,6 @@ def test_inplace_kv(self):
         actual = opt_func(k_cache.clone(), k.clone(), step)
         self.assertTrue(torch.allclose(expect.cpu(), actual.cpu()))
 
+
 if __name__ == "__main__":
     unittest.main()