From 6133b643d8c8d414584d09e6daf9a93896f5142a Mon Sep 17 00:00:00 2001 From: Mark Wiebe Date: Fri, 26 Oct 2012 15:33:42 -0700 Subject: [PATCH] Fix PTX lib detection logic on windows --- llvm-config-win32.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/llvm-config-win32.py b/llvm-config-win32.py index 0e70231..2370625 100644 --- a/llvm-config-win32.py +++ b/llvm-config-win32.py @@ -50,12 +50,12 @@ Advapi32 Shell32 """.split(): print('-l%s' % lib) - # Look for the PTX .lib in %LIBPATH% - if find_path_of('LLVMPTXCodeGen.lib', 'LIBPATH') != None: - print('-lLLVMPTXAsmPrinter') - print('-lLLVMPTXCodeGen') - print('-lLLVMPTXDesc') - print('-lLLVMPTXInfo') + llvmbin = find_path_of('llvm-tblgen.exe') + if os.path.exists(os.path.join(llvmbin, '../lib/LLVMPTXCodeGen.lib')): + print('-lLLVMPTXAsmPrinter') + print('-lLLVMPTXCodeGen') + print('-lLLVMPTXDesc') + print('-lLLVMPTXInfo') elif sys.argv[1] == '--includedir': llvmbin = find_path_of('llvm-tblgen.exe') if llvmbin is None: