@@ -20,6 +20,9 @@ def parse_args():
20
20
parser = argparse .ArgumentParser (description = """
21
21
Checks build arguments
22
22
""" )
23
+
24
+ default_gpus = 'gfx906:xnack-,gfx1030,gfx1100,gfx1101,gfx1102,gfx1151,gfx1200,gfx1201'
25
+
23
26
parser .add_argument ('-g' , '--debug' , required = False , default = False , action = 'store_true' ,
24
27
help = 'Generate Debug build (default: False)' )
25
28
parser .add_argument ( '--build_dir' , type = str , required = False , default = "build" ,
@@ -37,7 +40,7 @@ def parse_args():
37
40
help = 'Install after build (default: False)' )
38
41
parser .add_argument ( '--cmake-darg' , required = False , dest = 'cmake_dargs' , action = 'append' , default = [],
39
42
help = 'List of additional cmake defines for builds (e.g. CMAKE_CXX_COMPILER_LAUNCHER=ccache)' )
40
- parser .add_argument ('-a' , '--architecture' , dest = 'gpu_architecture' , required = False , default = "gfx906;gfx1030;gfx1100;gfx1101;gfx1102" , #:sramecc+:xnack-" ) #gfx1030" ) #gfx906" ) # gfx1030" )
43
+ parser .add_argument ('-a' , '--architecture' , dest = 'gpu_architecture' , required = False , default = default_gpus , #:sramecc+:xnack-" ) #gfx1030" ) #gfx906" ) # gfx1030" )
41
44
help = 'Set GPU architectures, e.g. all, gfx000, gfx803, gfx906:xnack-;gfx1030;gfx1100 (optional, default: all)' )
42
45
parser .add_argument ('-v' , '--verbose' , required = False , default = False , action = 'store_true' ,
43
46
help = 'Verbose build (default: False)' )
@@ -111,7 +114,7 @@ def config_cmd():
111
114
else :
112
115
cmake_executable = "cmake"
113
116
toolchain = "toolchain-linux.cmake"
114
- cmake_platform_opts = f"-DROCM_DIR:PATH={ rocm_path } -DCPACK_PACKAGING_INSTALL_PREFIX={ rocm_path } "
117
+ cmake_platform_opts = [ f"-DROCM_DIR:PATH={ rocm_path } " , f" -DCPACK_PACKAGING_INSTALL_PREFIX={ rocm_path } "]
115
118
116
119
tools = f"-DCMAKE_TOOLCHAIN_FILE={ toolchain } "
117
120
cmake_options .append ( tools )
0 commit comments