pass flags to transducer joint kernel (#1273)

c4e85f7b · Masaki Kozuki · GitHub · b2fdf9c4 · c4e85f7b
Unverified Commit c4e85f7b authored Jan 18, 2022 by Masaki Kozuki Committed by GitHub Jan 18, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

setup.py setup.py +2 -2

No files found.
--- a/setup.py
+++ b/setup.py
@@ -538,8 +538,8 @@ if "--transducer" in sys.argv:
                "apex/contrib/csrc/transducer/transducer_joint_kernel.cu",
            ],
            extra_compile_args={
-                "cxx": ["-O3"] + version_dependent_macros,
-                "nvcc": append_nvcc_threads(["-O3"] + version_dependent_macros),
+                "cxx": ["-O3"] + version_dependent_macros + generator_flag,
+                "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")],
        )