Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit 276a1fb7 authored by zhanghj2's avatar zhanghj2
Browse files

添加编译选项

parent ed798e2e
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 4 deletions
setup.py
View file @ 276a1fb7
...@@ -45,7 +45,7 @@ this_dir = os.path.dirname(os.path.abspath(__file__)) ...@@ -45,7 +45,7 @@ this_dir = os.path.dirname(os.path.abspath(__file__))
if False: if False:
cxx_args = ["/O2", "/std:c++20", "/DNDEBUG", "/W0"] cxx_args = ["/O2", "/std:c++20", "/DNDEBUG", "/W0"]
else: else:
cxx_args = ["-O3", "-std=c++17", "-DNDEBUG", "-Wno-deprecated-declarations", "-DDCU_ASM", "-Wno-return-type", ] cxx_args = ["-O3", "-std=c++20", "-DNDEBUG", "-Wno-deprecated-declarations", "-DDCU_ASM", "-Wno-return-type", ]
ext_modules = [] ext_modules = []
...@@ -96,7 +96,7 @@ ext_modules.append( ...@@ -96,7 +96,7 @@ ext_modules.append(
"cxx": cxx_args + get_features_args(), "cxx": cxx_args + get_features_args(),
"nvcc": [ "nvcc": [
"-O3", "-O3",
"-std=c++17", "-std=c++20",
"-DNDEBUG", "-DNDEBUG",
"-DHIP_ENABLE_WARP_SYNC_BUILTINS", "-DHIP_ENABLE_WARP_SYNC_BUILTINS",
"-ffast-math", "-ffast-math",
...@@ -105,8 +105,9 @@ ext_modules.append( ...@@ -105,8 +105,9 @@ ext_modules.append(
"-DDCU_ASM", "-DDCU_ASM",
"--save-temps", "--save-temps",
"-w", "-w",
"-mllvm", "-mllvm -enable-num-vgprs-512=true",
"-enable-num-vgprs-512=true", "-mllvm -allow-cse-cross-bb-convergent-call=true",
"-mllvm -full-vectorize-slp=true",
] + get_features_args() + get_arch_flags() ] + get_features_args() + get_arch_flags()
}, },
include_dirs=[ include_dirs=[
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment