File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -100,7 +100,7 @@ def wrapper(*args, **kwargs):
100100
101101def 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):
124124def 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 """
Original file line number Diff line number Diff 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+
147147def get_compute_capability ():
148148 if torch .cuda .is_available ():
149149 capability = torch .cuda .get_device_capability ()
You can’t perform that action at this time.
0 commit comments