diff --git a/test/network_ops/test_scatter_add.py b/test/network_ops/test_scatter_add.py index ef44ab23b008f156a34001e6c1c6436e2a556561..75ff03467152a70c16bd513f5b8b8d36915df34a 100644 --- a/test/network_ops/test_scatter_add.py +++ b/test/network_ops/test_scatter_add.py @@ -1,3 +1,4 @@ +import unittest import torch import numpy as np import torch_npu @@ -53,6 +54,7 @@ class TestScatterAdd(TestCase): npu_inp_output = self.npu_op_exec_inp(npu_input3, item[0], npu_input1, npu_input2) self.assertRtolEqual(cpu_inp_output, npu_inp_output) + @unittest.skip("skip test_scatter_add_float16_shape_format now") def test_scatter_add_float16_shape_format(self, device="npu"): def cpu_op_exec_inp_fp16(input1, dim, index, src): input1 = input1.to(torch.float32)