105105
106106MI300_ARCH = ("gfx940" , "gfx941" , "gfx942" )
107107NAVI_ARCH = ("gfx1030" , "gfx1100" , "gfx1101" , "gfx1200" , "gfx1201" )
108- < << << << HEAD
109108NAVI3_ARCH = ("gfx1100" , "gfx1101" )
110109NAVI4_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- > >> >> >> 3 b9e558b7c ([release / 2.6 ] skip convolution tests on Navi (#2055))
122118
123119def 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
19341929def 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- >> >> >> > 3 b9e558b7c ([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