You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
3 weeks ago
|
#!/usr/bin/python3
|
||
|
|
|
||
|
|
import os,sys,subprocess,math
|
||
|
|
from complib2 import *
|
||
|
|
from complib3 import gs_incremental_compile, gs_incremental_compile_list
|
||
|
|
|
||
|
|
import shutil
|
||
|
|
#from distutils.dir_util import copy_tree as copy_tree #this version does overwrites
|
||
|
|
from shutil import copytree
|
||
|
|
|
||
|
|
libname = 'amsculib2.linux64' #prefix static library name to generate
|
||
|
|
targetname = 'test' #create this executable when compiling tests
|
||
|
|
commonincdir = "../../linux64/include"
|
||
|
|
commonlibdir = "../../linux64/lib"
|
||
|
|
localbindir = "./bin_linux64"
|
||
|
|
cc = 'nvcc' #compiler
|
||
|
|
srcexts = ['.c','.cpp','.cu']
|
||
|
|
mainsrc = ['main.cu'] #ignore these files when compiling the static library
|
||
|
|
|
||
|
|
kwargs = dict()
|
||
|
|
include = "-I./include -I{}".format(commonincdir)
|
||
|
|
kwargs['include'] = include
|
||
|
|
#-dc flag: relocatable device code - needed for device functions to link in different "execution units"
|
||
|
|
#--ptxas-options=-v
|
||
|
|
kwargs['flags'] = "-dc --compiler-options '-fPIC -O3'"
|
||
|
|
kwargs['libdir'] = "-L{} -L{}".format(localbindir,commonlibdir)
|
||
|
|
kwargs['libflags'] = "-l{}".format(libname)
|
||
|
|
kwargs['linkerflags'] = ""
|
||
|
|
kwargs['recurse'] = True
|
||
|
|
kwargs['objstore'] = "./objstore"
|
||
|
|
kwargs['searchincdirs'] = ['./include']
|
||
|
|
|
||
|
|
#find all source files, except the main project files
|
||
|
|
files = flist('./src',exts = srcexts, recurse=True)
|
||
|
|
files = except_contains(files,mainsrc)
|
||
|
|
objfiles = replaceexts(files,'.o')
|
||
|
|
objfiles_sss = list_to_sss(objfiles)
|
||
|
|
|
||
|
|
#compile all the source files in the list
|
||
|
|
#gs_compile_list(cc,files,**kwargs)
|
||
|
|
gs_incremental_compile_list(cc,files,**kwargs)
|
||
|
|
|
||
|
|
#archive all the source files into a static library
|
||
|
|
#ar_list(objfiles,'{}/lib{}.a'.format(localbindir,libname))
|
||
|
|
objlist = flist(kwargs['objstore'],exts='.o',recurse=True)
|
||
|
|
ar_list(objlist,'{}/lib{}.a'.format(localbindir,libname))
|
||
|
|
|
||
|
|
# #Push any libraries to the common lib folder
|
||
|
|
shutil.copy('{}/lib{}.a'.format(localbindir,libname),commonlibdir)
|
||
|
|
|
||
|
|
# #Copy include files to the common include folder
|
||
|
|
copytree('./include/',commonincdir+'/',dirs_exist_ok=True)
|