mirror of https://github.com/Relintai/pmlpp.git
80 lines
2.1 KiB
Plaintext
80 lines
2.1 KiB
Plaintext
import os
|
|
|
|
Import('env')
|
|
|
|
module_env = env.Clone()
|
|
|
|
module_env.pmlpp_build_tests = True
|
|
|
|
if ARGUMENTS.get('pmlpp_build_tests', 'yes') == 'no':
|
|
module_env.pmlpp_build_tests = False
|
|
|
|
sources = [
|
|
"register_types.cpp",
|
|
|
|
"lin_alg/mlpp_vector.cpp",
|
|
"lin_alg/mlpp_matrix.cpp",
|
|
"lin_alg/mlpp_tensor3.cpp",
|
|
|
|
"activation/activation.cpp",
|
|
"ann/ann.cpp",
|
|
"auto_encoder/auto_encoder.cpp",
|
|
"bernoulli_nb/bernoulli_nb.cpp",
|
|
"c_log_log_reg/c_log_log_reg.cpp",
|
|
"convolutions/convolutions.cpp",
|
|
"cost/cost.cpp",
|
|
"data/data.cpp",
|
|
"dual_svc/dual_svc.cpp",
|
|
"exp_reg/exp_reg.cpp",
|
|
"gan/gan.cpp",
|
|
"gaussian_nb/gaussian_nb.cpp",
|
|
"gauss_markov_checker/gauss_markov_checker.cpp",
|
|
"hidden_layer/hidden_layer.cpp",
|
|
"hypothesis_testing/hypothesis_testing.cpp",
|
|
"kmeans/kmeans.cpp",
|
|
"knn/knn.cpp",
|
|
"lin_alg/lin_alg.cpp",
|
|
"lin_reg/lin_reg.cpp",
|
|
"log_reg/log_reg.cpp",
|
|
"mann/mann.cpp",
|
|
"mlp/mlp.cpp",
|
|
"multinomial_nb/multinomial_nb.cpp",
|
|
"multi_output_layer/multi_output_layer.cpp",
|
|
"numerical_analysis/numerical_analysis.cpp",
|
|
"outlier_finder/outlier_finder.cpp",
|
|
"output_layer/output_layer.cpp",
|
|
"pca/pca.cpp",
|
|
"probit_reg/probit_reg.cpp",
|
|
"regularization/reg.cpp",
|
|
"softmax_net/softmax_net.cpp",
|
|
"softmax_reg/softmax_reg.cpp",
|
|
"stat/stat.cpp",
|
|
"svc/svc.cpp",
|
|
"tanh_reg/tanh_reg.cpp",
|
|
"transforms/transforms.cpp",
|
|
"uni_lin_reg/uni_lin_reg.cpp",
|
|
"utilities/utilities.cpp",
|
|
"wgan/wgan.cpp",
|
|
]
|
|
|
|
if module_env.pmlpp_build_tests:
|
|
module_env.Prepend(CPPDEFINES=["TESTS_ENABLED"])
|
|
|
|
sources += [
|
|
"test/mlpp_tests.cpp",
|
|
"test/mlpp_matrix_tests.cpp",
|
|
]
|
|
|
|
|
|
if ARGUMENTS.get('pmlpp_shared', 'no') == 'yes':
|
|
# Shared lib compilation
|
|
module_env.Append(CCFLAGS=['-fPIC'])
|
|
module_env['LIBS'] = []
|
|
shared_lib = module_env.SharedLibrary(target='#bin/pmlpp', source=sources)
|
|
shared_lib_shim = shared_lib[0].name.rsplit('.', 1)[0]
|
|
env.Append(LIBS=[shared_lib_shim])
|
|
env.Append(LIBPATH=['#bin'])
|
|
else:
|
|
# Static compilation
|
|
module_env.add_source_files(env.modules_sources, sources)
|