use nvfuser_codegen

NVIDIA · Apr 19, 2023 · 0da3ffb · 0da3ffb
1 parent 7642c1c
commit 0da3ffb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -373,7 +373,7 @@ def check_cudnn_version_and_warn(global_option: str, required_cudnn_version: int
     if nvfuser_is_refactored:
         include_dirs.append(os.path.join(PYTORCH_HOME, "third_party/nvfuser/csrc"))
         library_dirs = nvfuser.__path__
-        extra_link_args.append("-lnvfuser")
+        extra_link_args.append("-lnvfuser_codegen")
     ext_modules.append(
         CUDAExtension(
             name='instance_norm_nvfuser_cuda',