Skip to content

GitLab

"git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "29aea6a206d5ecfec1e7b0addab1792a0833e47e"
Commit 39ed597b authored by Tri Dao's avatar Tri Dao
Browse files

[LayerNorm] Compile for both sm70 and sm80

parent 71f674ae
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
csrc/layer_norm/setup.py
View file @ 39ed597b
...@@ -98,8 +98,8 @@ if os.path.exists(os.path.join(torch_dir, "include", "ATen", "CUDAGeneratorImpl. ...@@ -98,8 +98,8 @@ if os.path.exists(os.path.join(torch_dir, "include", "ATen", "CUDAGeneratorImpl.
raise_if_cuda_home_none("--fast_layer_norm") raise_if_cuda_home_none("--fast_layer_norm")
# Check, if CUDA11 is installed for compute capability 8.0 # Check, if CUDA11 is installed for compute capability 8.0
cc_flag = [] cc_flag = []
# cc_flag.append("-gencode") cc_flag.append("-gencode")
# cc_flag.append("arch=compute_70,code=sm_70") cc_flag.append("arch=compute_70,code=sm_70")
cc_flag.append("-gencode") cc_flag.append("-gencode")
cc_flag.append("arch=compute_80,code=sm_80") cc_flag.append("arch=compute_80,code=sm_80")
......
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