Fix PTX lib detection logic on windows
This commit is contained in:
parent
daa62d9f78
commit
6133b643d8
1 changed files with 6 additions and 6 deletions
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue