def initialize_asp_mod(self):
# Create ASP module
GMM.asp_mod = asp_module.ASPModule(use_cuda=GMM.use_cuda, use_cilk=GMM.use_cilk, use_tbb=GMM.use_tbb)
if GMM.use_cuda:
self.insert_base_code_into_listed_modules(['c++'])
self.insert_non_rendered_code_into_cuda_module()
self.insert_rendered_code_into_module('cuda')
GMM.asp_mod.backends['cuda'].toolchain.cflags.extend(["-Xcompiler","-fPIC","-arch=sm_%s%s" % GMM.platform_info['cuda']['capability'] ])
GMM.asp_mod.backends['c++'].compilable = False # TODO: For now, must force ONLY cuda backend to compile
if GMM.use_cilk:
self.insert_base_code_into_listed_modules(['cilk'])
self.insert_non_rendered_code_into_cilk_module()
self.insert_rendered_code_into_module('cilk')
GMM.asp_mod.backends['cilk'].toolchain.cc = 'icc'
GMM.asp_mod.backends['cilk'].toolchain.cflags = ['-O2','-gcc', '-ip','-fPIC']
if GMM.use_tbb:
self.insert_base_code_into_listed_modules(['tbb'])
self.insert_non_rendered_code_into_tbb_module()
self.insert_rendered_code_into_module('tbb')
# Setup toolchain
from codepy.libraries import add_numpy, add_boost_python, add_cuda
for name, mod in GMM.asp_mod.backends.iteritems():
add_numpy(mod.toolchain)
add_boost_python(mod.toolchain)
if name in ['cuda']:
add_cuda(mod.toolchain)
return GMM.asp_mod
def initialize_asp_mod(self):
# Create ASP module
GMM.asp_mod = asp_module.ASPModule(use_cuda=GMM.use_cuda, use_cilk=GMM.use_cilk)
if GMM.use_cuda:
print "GMM SPECIALIZER: USING CUDA"
self.insert_base_code_into_listed_modules(['c++'])
self.insert_non_rendered_code_into_cuda_module()
self.insert_rendered_code_into_module('cuda')
GMM.asp_mod.backends['cuda'].toolchain.cflags.extend(["-Xcompiler","-fPIC","-arch=sm_%s%s" % GMM.platform_info['cuda']['capability'] ])
# TODO: For now, must force ONLY cuda backend to compile
GMM.asp_mod.backends['c++'].compilable = False
#print GMM.asp_mod.generate()
if GMM.use_cilk:
print "GMM SPECIALIZER: USING CILK+"
self.insert_base_code_into_listed_modules(['cilk'])
self.insert_non_rendered_code_into_cilk_module()
self.insert_rendered_code_into_module('cilk')
GMM.asp_mod.backends['cilk'].toolchain.cc = 'icc'
GMM.asp_mod.backends['cilk'].toolchain.cflags = ['-O2','-gcc', '-ip','-fPIC']
# Setup toolchain
from codepy.libraries import add_numpy, add_boost_python, add_cuda
for name, mod in GMM.asp_mod.backends.iteritems():
add_numpy(mod.toolchain)
add_boost_python(mod.toolchain)
if name in ['cuda']:
add_cuda(mod.toolchain)
return GMM.asp_mod
def initialize_asp_mod(self):
# Create ASP module
SVM.asp_mod = asp_module.ASPModule(use_cuda=SVM.use_cuda)
if SVM.use_cuda:
self.insert_cache_controller_code_into_listed_modules(['c++', 'cuda'])
self.insert_base_code_into_listed_modules(['c++'])
self.insert_non_rendered_code_into_module()
self.insert_rendered_code_into_cuda_module()
SVM.asp_mod.backends['cuda'].toolchain.cflags.extend(["-Xcompiler","-fPIC", "-arch=sm_%s%s" % SVM.platform_info['cuda']['capability'] ])
SVM.asp_mod.backends['c++'].toolchain.cflags.extend(["-lcublas"])
SVM.asp_mod.backends['c++'].compilable = False # TODO: For now, must force ONLY cuda backend to compile
#print SVM.asp_mod.generate()
# Setup toolchain
for name, mod in SVM.asp_mod.backends.iteritems():
add_numpy(mod.toolchain)
add_boost_python(mod.toolchain)
if name in ['cuda']:
add_cuda(mod.toolchain)
return SVM.asp_mod
def initialize_asp_mod(self):
# Create ASP module
SVM.asp_mod = asp_module.ASPModule(use_cuda=SVM.use_cuda)
if SVM.use_cuda:
self.insert_cache_controller_code_into_listed_modules(['c++', 'cuda'])
self.insert_base_code_into_listed_modules(['c++'])
self.insert_non_rendered_code_into_module()
self.insert_rendered_code_into_cuda_module()
SVM.asp_mod.backends['cuda'].toolchain.cflags.extend(["-Xcompiler","-fPIC", "-arch=sm_%s%s" % SVM.platform_info['cuda']['capability'] ])
SVM.asp_mod.backends['c++'].toolchain.cflags.extend(["-lcublas"])
SVM.asp_mod.backends['c++'].compilable = False # TODO: For now, must force ONLY cuda backend to compile
#print SVM.asp_mod.generate()
# Setup toolchain
for name, mod in SVM.asp_mod.backends.iteritems():
add_numpy(mod.toolchain)
add_boost_python(mod.toolchain)
if name in ['cuda']:
add_cuda(mod.toolchain)
return SVM.asp_mod
