Skip to content

Commit aab4cc6

Browse files
committed
fix format issue
1 parent 1847f3e commit aab4cc6

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

torchao/testing/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ def wrapper(*args, **kwargs):
100100

101101
def skip_if_no_xpu(message=None):
102102
"""Decorator to skip tests on ROCm platform with custom message.
103-
103+
104104
Args:
105105
message (str, optional): Additional information about why the test is skipped.
106106
"""
@@ -124,7 +124,7 @@ def wrapper(*args, **kwargs):
124124
def skip_if_xpu(message=None):
125125
"""
126126
Decorator to skip tests if XPU is available.
127-
127+
128128
Args:
129129
message (str, optional): Additional information about why the test is skipped.
130130
"""

torchao/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,8 +142,8 @@ def auto_detect_device():
142142
return torch.accelerator.current_accelerator()
143143
else:
144144
return None
145-
146-
145+
146+
147147
def get_compute_capability():
148148
if torch.cuda.is_available():
149149
capability = torch.cuda.get_device_capability()

0 commit comments

Comments
 (0)