Mentions légales du service

Skip to content
Snippets Groups Projects
Commit 2aede5ce authored by hhakim's avatar hhakim
Browse files

Add test MEG_factorization (GPU2, FaustGPU and CPU).

parent 432e773c
Branches
Tags
No related merge requests found
Hide whitespace changes
Inline Side-by-side
misc/test/CMakeLists.txt
+ 5
− 4
View file @ 2aede5ce
......@@ -156,7 +156,7 @@ endif()
if(NOT NOCPPTESTS)
foreach(TEST_FPP float double)
foreach(FILE faust_mult2 faust_mult faust_mult_cplx test_Vect_min test_MatDense_get_row test_MatDense_lower_upper_tri test_MatDense_nonzeros_indices test_Transform_move test_TransformHelper_and_Transform_copy_ctor test_TransformHelper_and_Transform_fac_iterato test_TransformHelper_variadic_template_ctor test_MatDense_min test_TH_pack_factors faust_transform_omp_mul faust_pruneout faust_transform_optimize_storage faust_transform_optimize faust_prox_blockdiag)
foreach(FILE faust_mult2 faust_mult faust_mult_cplx test_Vect_min test_MatDense_get_row test_MatDense_lower_upper_tri test_MatDense_nonzeros_indices test_Transform_move test_TransformHelper_and_Transform_copy_ctor test_TransformHelper_and_Transform_fac_iterato test_TransformHelper_variadic_template_ctor test_MatDense_min test_TH_pack_factors faust_transform_omp_mul faust_pruneout faust_transform_optimize_storage faust_transform_optimize faust_prox_blockdiag)
if(USE_GPU_MOD AND ${FILE} MATCHES test_TH_pack_factors|test_TransformHelper_variadic_template_ctor|faust_transform_optimize|test_TransformHelper_and_Transform_fac_iterato|faust_mult2|faust_transform_omp_mul|test_Transform_move|test_TransformHelper_and_Transform_copy_ctor AND ${TEST_FPP} MATCHES float)# gpu_mod handles float but not FaustGPU
message(STATUS ${FILE}_${TEST_FPP} " skipped")
continue()
......@@ -205,20 +205,20 @@ if(MATIO_LIB_FILE AND MATIO_INC_DIR AND BUILD_READ_MAT_FILE AND NOT NOCPPTESTS)
# faust_multiplication : time comparison between Faust-vector product and Dense matrix-vector product
list(APPEND tests hierarchicalFactorization hierarchicalFactorizationFFT test_palm4MSA test_palm4MSAFFT faust_multiplication faust_matdense_conjugate GivensFGFT GivensFGFTSparse GivensFGFTParallel GivensFGFTParallelSparse test_MatDiag faust_matsparse_mul faust_matsparse_index_op GivensFGFTComplex GivensFGFTComplexSparse GivensFGFTParallelComplex faust_toeplitz faust_circ faust_hankel palm4msa_2020 hierarchical2020 hierarchical2020Hadamard hierarchicalFactorizationHadamard)#MEG_factorization)
list(APPEND tests hierarchicalFactorization hierarchicalFactorizationFFT test_palm4MSA test_palm4MSAFFT faust_multiplication faust_matdense_conjugate GivensFGFT GivensFGFTSparse GivensFGFTParallel GivensFGFTParallelSparse test_MatDiag faust_matsparse_mul faust_matsparse_index_op GivensFGFTComplex GivensFGFTComplexSparse GivensFGFTParallelComplex faust_toeplitz faust_circ faust_hankel palm4msa_2020 hierarchical2020 hierarchical2020Hadamard hierarchicalFactorizationHadamard)
if(FAUST_TORCH)
list(APPEND tests faust_torch)
endif()
if(USE_GPU_MOD)
list(APPEND tests faust_gpu_mod hierarchical2020_gpu test_matdense_gpu_mod test_matsparse_gpu_mod test_transform_gpu_mod test_vect_gpu_mod test_transform_helper_gpu_mod hierarchical2020_gpu2 hierarchical2020Hadamard_gpu2)
list(APPEND tests faust_gpu_mod hierarchical2020_gpu test_matdense_gpu_mod test_matsparse_gpu_mod test_transform_gpu_mod test_vect_gpu_mod test_transform_helper_gpu_mod hierarchical2020_gpu2 hierarchical2020Hadamard_gpu2 MEG_factorization)
endif()
foreach(TEST_FPP float double complex<float> complex<double>)
foreach(testin IN LISTS tests)
if(USE_GPU_MOD AND (${testin} MATCHES gpu|hierarchical2020Hadamard|palm4msa_2020|hierarchical2020|hierarchicalFactorizationHadamard) AND ${TEST_FPP} MATCHES float ) # gpu_mod handles float but not FaustGPU
if(USE_GPU_MOD AND (${testin} MATCHES gpu|hierarchical2020Hadamard|palm4msa_2020|hierarchical2020|hierarchicalFactorizationHadamard|MEG_factorization) AND ${TEST_FPP} MATCHES float ) # gpu_mod handles float but not FaustGPU
message(STATUS "${testin}_${TEST_FPP} skipped")
continue()
endif()
......@@ -386,6 +386,7 @@ if(NOT NOCPPTESTS)
endif(BUILD_READ_XML_FILE)
add_executable(test_RefManager ${FAUST_SRC_TEST_SRC_CPP_DIR}/test_RefManager.cpp ${FAUST_BIN_TEST_SRC_DIR}/)
target_link_libraries(test_RefManager ${FAUST_TARGET} ${MATIO_LIB_FILE} ${HDF5_LIB_FILE} ${OPENBLAS_LIB_FILE})
add_test(NAME test_RefManager COMMAND ${FAUST_BIN_TEST_BIN_DIR}/test_RefManager)
......
misc/test/src/C++/MEG_factorization.cpp.in 0 → 100644
+ 144
− 0
View file @ 2aede5ce
#include "faust_init_from_matio_params.h"
#include "faust_init_from_matio_core.h"
#include "faust_TransformHelper.h"
#include "faust_HierarchicalFact.h"
#include "faust_ConstraintInt.h"
#include "faust_MatDense.h"
#include "faust_hierarchical.h"
#ifdef USE_GPU_MOD
#include"faust_gpu_mod_utils.h"
#include "faust_gpu_mod.h"
#endif
#include <vector>
#include <cstdlib>
using namespace Faust;
using namespace std;
#ifdef ON_CPU
const FDevice dev = Cpu;
#else
const FDevice dev = GPU2;
#endif
int main(int argc, const char** argv)
{
#if(defined(USE_GPU_MOD))
Faust::enable_gpu_mod();
#endif
int impl = 2016;
bool on_gpu = false;
if(argc > 1)
{
impl = atoi(argv[1]);
if(impl != 2016 && impl != 2020)
{
throw runtime_error("Implementation (1st arg.) must be 2016 or 2020.");
}
if(argc > 2)
{
string arg2 = argv[1];
on_gpu = true;
}
}
// CONSTRAINT_NAME_SPLINCOL
// CONSTRAINT_NAME_SP
// const string MatrixFilename = "@FAUST_DATA_MAT_DIR@/matrix_MEG.mat";
const string MatrixFilename = "matrix_MEG.mat";
Faust::MatDense<double,Cpu> MEG;
#if(!defined(ON_CPU) && defined(USE_GPU_MOD))
Faust::MatDense<double,GPU2> gpu_MEG;
#endif
init_faust_mat_from_matio(MEG, MatrixFilename.c_str(), "matrix");
MEG.transpose();
#if(!defined(ON_CPU) && defined(USE_GPU_MOD))
if(dev == GPU2)
gpu_MEG = MEG;
#endif
vector<const Faust::ConstraintGeneric*> fac_constraints;
vector<const Faust::ConstraintGeneric*> res_constraints;
vector<unsigned int> res_cons_pvals = {58263, 46610, 37288, 29831, 23865, 19092, 15274, 12219};
Faust::ConstraintInt<double,dev> cons_i(
CONSTRAINT_NAME_SP,
1632,
204,
204);
Faust::ConstraintInt<double,dev> cons_1(
CONSTRAINT_NAME_SPCOL,
10,
204,
8193);
fac_constraints.push_back(&cons_1);
for(int i=0;i<7;i++)
fac_constraints.push_back(&cons_i);
for(int i=0;i<8;i++)
{
auto res_cons = new Faust::ConstraintInt<double, dev>(CONSTRAINT_NAME_SP, res_cons_pvals[i], MEG.getNbRow(), MEG.getNbRow());
res_constraints.push_back(res_cons);
}
vector<vector<const Faust::ConstraintGeneric*>> constraints = { res_constraints, fac_constraints };
Faust::StoppingCriterion<double> loc_opt_stop(10);
Faust::StoppingCriterion<double> glob_opt_stop(10);
Faust::Params<double, dev> params(MEG.getNbRow(), MEG.getNbCol(), 9, constraints, {}, loc_opt_stop, glob_opt_stop, /*verbosity*/ false, /* isUpdateWayR2L_ */ true, /* isFactSideLeft_ */ true);
Faust::BlasHandle<Cpu> cublasHandle;
Faust::SpBlasHandle<Cpu> cusparseHandle;
double lambda2020;
char* str_use_csr = getenv("USE_CSR");
char* str_packing_RL = getenv("PACKING_RL");
char* str_isUpdateWayR2L = getenv("UR2L");
char* str_isFactSideLeft = getenv("FSL");
params.isUpdateWayR2L = false;
params.isFactSideLeft = true;
params.packing_RL = true;
params.use_csr = true;
if(str_packing_RL)
params.packing_RL = std::atoi(str_packing_RL) != 0;
if(str_use_csr)
params.use_csr = std::atoi(str_use_csr) != 0;
if(str_isUpdateWayR2L)
params.isUpdateWayR2L = std::atoi(str_isUpdateWayR2L) != 0;
if(str_isFactSideLeft)
params.isFactSideLeft = std::atoi(str_isFactSideLeft) != 0;
params.Display();
if(impl == 2016 || argc < 2)
{
// Faust::HierarchicalFact<double, dev> algo(MEG, params, cublasHandle, cusparseHandle);
// algo.compute_facts();
}
else if(impl == 2020)
{
//Faust::TransformHelper<FPP,DEVICE>* Faust::hierarchical(const Faust::MatDense<FPP,DEVICE>& A,
// Params<FPP,DEVICE, Real<FPP>> & p,
// Real<FPP>& lambda, const bool compute_2norm_on_array,
// const bool on_gpu)
#ifdef ON_CPU
cout << "on_gpu: " << on_gpu << endl;
auto th = Faust::hierarchical<double,Cpu>(MEG, params, lambda2020, false, on_gpu);
#else
cout << "hierarchical on GPU2 MEG" << endl;
auto th = Faust::hierarchical<double,GPU2>(gpu_MEG, params, lambda2020, false, on_gpu);
#endif
th->display();
delete th;
}
for(auto c: res_constraints)
{
delete c;
}
return EXIT_SUCCESS;
}
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment