Skip to content

GitLab

"integration-tests/vscode:/vscode.git/clone" did not exist on "e36dfaa8de2d9a9fa67eeed5ce64fd5949916c99"
Unverified Commit 47c269b6 authored by Masaki Kozuki's avatar Masaki Kozuki Committed by GitHub
Browse files

build fused grad accum w/ wgrad only if cuda>10 (#1312)

parent ddc08039
Show whitespace changes
Inline Side-by-side
Showing with 28 additions and 28 deletions
setup.py
View file @ 47c269b6
...@@ -298,9 +298,9 @@ if "--cuda_ext" in sys.argv: ...@@ -298,9 +298,9 @@ if "--cuda_ext" in sys.argv:
) )
# Check, if CUDA11 is installed for compute capability 8.0 # Check, if CUDA11 is installed for compute capability 8.0
cc_flag = []
_, bare_metal_major, bare_metal_minor = get_cuda_bare_metal_version(CUDA_HOME) _, bare_metal_major, bare_metal_minor = get_cuda_bare_metal_version(CUDA_HOME)
if int(bare_metal_major) >= 11: if int(bare_metal_major) >= 11:
cc_flag = []
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")
if int(bare_metal_minor) > 0: if int(bare_metal_minor) > 0:
......
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