Skip to content

Commit a6b4789

Browse files
committed
refine code
1 parent 2db0d70 commit a6b4789

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

test/integration/test_integration.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1263,11 +1263,7 @@ class SmoothquantIntegrationTest(unittest.TestCase):
12631263
@unittest.skipIf(not torch.accelerator.is_available(), "Need GPU available")
12641264
@unittest.skip("Seg fault?")
12651265
def test_non_dynamically_quantizable_linear(self):
1266-
if (
1267-
torch.cuda.is_available()
1268-
and torch.cuda.is_available()
1269-
and torch.cuda.get_device_capability() < (8, 0)
1270-
):
1266+
if torch.cuda.is_available() and torch.cuda.get_device_capability() < (8, 0):
12711267
self.skipTest("test requires SM capability of at least (8, 0).")
12721268
model = (
12731269
torch.nn.Sequential(

0 commit comments

Comments
 (0)