Skip to content

GitLab

Commit 06612d65 authored by zhanghj2's avatar zhanghj2
Browse files

便于测试和出包

parent 2033d805
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
setup.py
View file @ 06612d65
...@@ -24,7 +24,7 @@ def get_features_args(): ...@@ -24,7 +24,7 @@ def get_features_args():
def get_arch_flags(): def get_arch_flags():
arch_flags = [] arch_flags = []
arch_flags.append("--offload-arch=gfx938") arch_flags.append("--offload-arch=gfx938;gfx936")
return arch_flags return arch_flags
# def get_nvcc_thread_args(): # def get_nvcc_thread_args():
...@@ -92,8 +92,7 @@ ext_modules.append( ...@@ -92,8 +92,7 @@ ext_modules.append(
Path(this_dir) / "csrc", Path(this_dir) / "csrc",
Path(this_dir) / "csrc" / "kerutils" / "include", # TODO Remove me Path(this_dir) / "csrc" / "kerutils" / "include", # TODO Remove me
Path(this_dir) / "csrc" / "sm90", Path(this_dir) / "csrc" / "sm90",
"/public/home/zhanghj/work/dev/cutlass_3.2.1-mla/include", Path(this_dir) / "csrc" / "cutlass" / "include"
# "/public/home/zhanghj/work/dev/cutlass_3.2.1-mla/tools/util/include",
], ],
) )
) )
......
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