From 519fd3cee0f89539b085bda6b1fa24749dfa1817 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=88=9A=E4=BD=99=E8=88=AA?= <qiyuhang@bytedance.com>
Date: Wed, 31 Dec 2025 16:59:27 +0800
Subject: [PATCH] Link libcuda.so

---
 hopper/setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/hopper/setup.py b/hopper/setup.py
index 95729edabe2..faa26e7958f 100644
--- a/hopper/setup.py
+++ b/hopper/setup.py
@@ -625,6 +625,7 @@ def nvcc_threads_args():
                 "cxx": ["-O3", "-std=c++17", "-DPy_LIMITED_API=0x03090000"] + stable_args + feature_args,
                 "nvcc": nvcc_threads_args() + nvcc_flags + cc_flag + feature_args,
             },
+            extra_link_args=["-lcuda"],
             include_dirs=include_dirs,
             py_limited_api=True,
         )