Skip to content

Commit 7808297

Browse files
committed
merge NPU(Ascend) backend logic in Int4PlainInt32Tensor subclass
1 parent 91cb46e commit 7808297

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/quantization/quantize_/workflows/int4/test_int4_plain_int32_tensor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,8 +100,8 @@ def test_activation_prescaling(self):
100100

101101
@unittest.skipIf(not torch_version_at_least("2.7.1"), "Need pytorch 2.7.1+")
102102
@unittest.skipIf(
103-
torch.accelerator.current_accelerator().type == "npu"
104-
and torch.accelerator.is_available(),
103+
torch.accelerator.current_accelerator().type != "npu"
104+
or not torch.accelerator.is_available(),
105105
"NPU not available",
106106
)
107107
class Int4PlainInt32TensorNPU(TestCase):

0 commit comments

Comments
 (0)