From f6d42f2dd01779acfddf1b8db2b05fd30f49c5b3 Mon Sep 17 00:00:00 2001
From: "q.yao" <streetyao@live.com>
Date: Thu, 15 Jun 2023 00:09:12 +0800
Subject: [PATCH] add library_dirs (#977)

---
 python/setup.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/python/setup.py b/python/setup.py
index a87c82f4..266228d1 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -108,10 +108,15 @@ include_dirs = [
 ]
 
 
+library_dirs = [
+    cuda_install_path + '/lib64',
+]
+
 ext_modules = [
     Pybind11Extension('cutlass_bindings',
                       ['cutlass/cpp/cutlass_bindings.cpp'],
                       include_dirs=include_dirs,
+                      library_dirs=library_dirs,
                       extra_compile_args=['-Xcompiler="-fpermissive"', '-w', '-std=c++17'],
                       libraries=['cudart'])
 ]