Fix PTX lib detection logic on windows

This commit is contained in:
Mark Wiebe 2012-10-26 15:33:42 -07:00
commit 6133b643d8

View file

@ -50,12 +50,12 @@ Advapi32
Shell32 Shell32
""".split(): """.split():
print('-l%s' % lib) print('-l%s' % lib)
# Look for the PTX .lib in %LIBPATH% llvmbin = find_path_of('llvm-tblgen.exe')
if find_path_of('LLVMPTXCodeGen.lib', 'LIBPATH') != None: if os.path.exists(os.path.join(llvmbin, '../lib/LLVMPTXCodeGen.lib')):
print('-lLLVMPTXAsmPrinter') print('-lLLVMPTXAsmPrinter')
print('-lLLVMPTXCodeGen') print('-lLLVMPTXCodeGen')
print('-lLLVMPTXDesc') print('-lLLVMPTXDesc')
print('-lLLVMPTXInfo') print('-lLLVMPTXInfo')
elif sys.argv[1] == '--includedir': elif sys.argv[1] == '--includedir':
llvmbin = find_path_of('llvm-tblgen.exe') llvmbin = find_path_of('llvm-tblgen.exe')
if llvmbin is None: if llvmbin is None: