Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down
10 changes: 6 additions & 4 deletions tests/py/ts/api/test_classes.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,21 +9,21 @@

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

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(
Expand Down Expand Up @@ -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
Expand Down
Loading