Skip to content

Commit 3f3b83a

Browse files
authored
Update common_utils.py
Fix merge conflicts
1 parent 21fcf59 commit 3f3b83a

File tree

1 file changed

+1
-13
lines changed

1 file changed

+1
-13
lines changed

torch/testing/_internal/common_utils.py

Lines changed: 1 addition & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,6 @@
105105

106106
MI300_ARCH = ("gfx940", "gfx941", "gfx942")
107107
NAVI_ARCH = ("gfx1030", "gfx1100", "gfx1101", "gfx1200", "gfx1201")
108-
<<<<<<< HEAD
109108
NAVI3_ARCH = ("gfx1100", "gfx1101")
110109
NAVI4_ARCH = ("gfx1200", "gfx1201")
111110

@@ -116,9 +115,6 @@ def is_navi3_arch():
116115
if gfx_arch in NAVI3_ARCH:
117116
return True
118117
return False
119-
=======
120-
NAVI4_ARCH = ("gfx1200", "gfx1201")
121-
>>>>>>> 3b9e558b7c ([release/2.6] skip convolution tests on Navi (#2055))
122118

123119
def freeze_rng_state(*args, **kwargs):
124120
return torch.testing._utils.freeze_rng_state(*args, **kwargs)
@@ -1930,19 +1926,11 @@ def wrapper(*args, **kwargs):
19301926
return dec_fn(func)
19311927
return dec_fn
19321928

1933-
<<<<<<< HEAD
19341929
def skipIfRocmArch(arch: tuple[str, ...]):
19351930
def dec_fn(fn):
19361931
@wraps(fn)
19371932
def wrap_fn(self, *args, **kwargs):
19381933
if TEST_WITH_ROCM:
1939-
=======
1940-
def skipIfRocmArch(arch: Tuple[str, ...]):
1941-
def dec_fn(fn):
1942-
@wraps(fn)
1943-
def wrap_fn(self, *args, **kwargs):
1944-
if TEST_WITH_ROCM: # noqa: F821
1945-
>>>>>>> 3b9e558b7c ([release/2.6] skip convolution tests on Navi (#2055))
19461934
prop = torch.cuda.get_device_properties(0)
19471935
if prop.gcnArchName.split(":")[0] in arch:
19481936
reason = f"skipIfRocm: test skipped on {arch}"
@@ -5772,4 +5760,4 @@ def wrapper(self, *args, **kwargs):
57725760
setattr(self, member, original_value)
57735761

57745762
return wrapper
5775-
return decorator
5763+
return decorator

0 commit comments

Comments
 (0)