Skip to content

GitLab

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