From 194b0043a9362e64e2de57c90f5e1e43805e0c7d Mon Sep 17 00:00:00 2001
From: Maxime <672982+maximegmd@users.noreply.github.com>
Date: Wed, 30 Aug 2023 19:37:16 +0200
Subject: [PATCH] fix: linker issues in conda environments #3929 (#4235)

* fix: linker issues in conda environments #3929

* ignore: re-ordering

* Update builder.py
---
 op_builder/builder.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/op_builder/builder.py b/op_builder/builder.py
index 9747261b..1fdcd485 100644
--- a/op_builder/builder.py
+++ b/op_builder/builder.py
@@ -702,6 +702,8 @@ class TorchCPUOpBuilder(CUDAOpBuilder):
         if not self.build_for_cpu:
             if not self.is_rocm_pytorch():
                 CUDA_LIB64 = os.path.join(torch.utils.cpp_extension.CUDA_HOME, "lib64")
+                if not os.path.exists(CUDA_LIB64):
+                    CUDA_LIB64 = os.path.join(torch.utils.cpp_extension.CUDA_HOME, "lib")
             else:
                 CUDA_LIB64 = os.path.join(torch.utils.cpp_extension.ROCM_HOME, "lib")
 
-- 
GitLab