Skip to content

GitLab

Commit 5d62c0d7 authored by zhanghj2's avatar zhanghj2
Browse files

938架构

parent 7a8722d7
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 5 deletions
setup.py
View file @ 5d62c0d7
...@@ -23,17 +23,19 @@ def get_features_args(): ...@@ -23,17 +23,19 @@ def get_features_args():
return features_args return features_args
def get_arch_flags(): def get_arch_flags():
return ["--offload-arch=gfx938"] arch_flags = []
arch_flags.append("--offload-arch=gfx938")
return arch_flags
def get_nvcc_thread_args(): # def get_nvcc_thread_args():
# nvcc_threads = os.getenv("NVCC_THREADS") or "32" # # nvcc_threads = os.getenv("NVCC_THREADS") or "32"
return ["--threads", nvcc_threads] # return ["--threads", nvcc_threads]
# subprocess.run(["git", "submodule", "update", "--init", "csrc/cutlass"]) # subprocess.run(["git", "submodule", "update", "--init", "csrc/cutlass"])
this_dir = os.path.dirname(os.path.abspath(__file__)) this_dir = os.path.dirname(os.path.abspath(__file__))
if IS_WINDOWS: 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++17", "-DNDEBUG", "-Wno-deprecated-declarations", "-DDCU_ASM", "-Wno-return-type", ]
......
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