use nvfuser_codegen

NVIDIA · Feb 12, 2023 · 5d4b5e3 · 5d4b5e3
1 parent a169c51
commit 5d4b5e3
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',