Skip to content

GitLab

"src/vscode:/vscode.git/clone" did not exist on "3c133f818b7dd51e2ee2ec8f68e5211de736f2f2"
Unverified Commit c4e85f7b authored by Masaki Kozuki's avatar Masaki Kozuki Committed by GitHub
Browse files

pass flags to transducer joint kernel (#1273)

parent b2fdf9c4
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
setup.py
View file @ c4e85f7b
...@@ -538,8 +538,8 @@ if "--transducer" in sys.argv: ...@@ -538,8 +538,8 @@ if "--transducer" in sys.argv:
"apex/contrib/csrc/transducer/transducer_joint_kernel.cu", "apex/contrib/csrc/transducer/transducer_joint_kernel.cu",
], ],
extra_compile_args={ extra_compile_args={
"cxx": ["-O3"] + version_dependent_macros, "cxx": ["-O3"] + version_dependent_macros + generator_flag,
"nvcc": append_nvcc_threads(["-O3"] + version_dependent_macros), "nvcc": append_nvcc_threads(["-O3"] + version_dependent_macros + generator_flag),
}, },
include_dirs=[os.path.join(this_dir, "csrc"), os.path.join(this_dir, "apex/contrib/csrc/multihead_attn")], include_dirs=[os.path.join(this_dir, "csrc"), os.path.join(this_dir, "apex/contrib/csrc/multihead_attn")],
) )
......
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