From dada62b7dfcd3a507b8e51bd225bade8f267a817 Mon Sep 17 00:00:00 2001 From: Joseph Loftin Date: Mon, 27 Jul 2026 20:52:38 +0000 Subject: [PATCH] Fix Blackwell SM --- pyproject.toml | 5 +++++ tests/py/ts/api/test_classes.py | 10 ++++++---- 2 files changed, 11 insertions(+), 4 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 88d9bbc743..7eec4836cb 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -468,6 +468,11 @@ extend-ignore-identifiers-re = [ "^([A-z]|[a-z])*nd*", "activ*([A-z]|[a-z]|[0-9])*,", ] +# Honor spellchecker:disable-line and spellchecker:off/:on regions. +extend-ignore-re = [ + "(?Rm)^.*(#|//)\\s*spellchecker:disable-line$", + "(?s)(#|//)\\s*spellchecker:off.*?(#|//)\\s*spellchecker:on", +] [tool.typos.default.extend-words] arange = "arange" diff --git a/tests/py/ts/api/test_classes.py b/tests/py/ts/api/test_classes.py index 557b05a1ff..899540359a 100644 --- a/tests/py/ts/api/test_classes.py +++ b/tests/py/ts/api/test_classes.py @@ -9,12 +9,12 @@ def is_blackwell(): """ - Check if running on NVIDIA Blackwell architecture (sm_90+). + Check if running on NVIDIA Blackwell architecture (sm_100+). Blackwell architecture adds input/output reformat layers in TensorRT engines. Returns: - bool: True if running on Blackwell (sm_90+), False otherwise + bool: True if running on Blackwell (sm_100+), False otherwise """ if not torch.cuda.is_available(): return False @@ -22,8 +22,8 @@ def is_blackwell(): device_properties = torch.cuda.get_device_properties(0) compute_capability = device_properties.major * 10 + device_properties.minor - # Blackwell is sm_90 and above - return compute_capability >= 90 + # Blackwell is sm_100 and above + return compute_capability >= 100 @unittest.skipIf( @@ -358,11 +358,13 @@ def test_get_layer_info(self): import json if is_blackwell(): + # spellchecker:off # blackwell has additional layers- # Layer 0: __mye88_myl0_0 ← Input reformat layer # Layer 1: aten__matmul(...) fc1 ← First matmul (fc1) # Layer 2: aten__matmul(...) fc2 ← Second matmul (fc2) # Layer 3: __mye90_myl0_3 ← Output reformat layer + # spellchecker:on num_layers = 4 else: num_layers = 2