ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
|
#include "../kernels/math_kernel_op.h"
#include "../para_gemm.h"
#include <gtest/gtest.h>
#include <iostream>
#include <mpi.h>
#include <vector>
Classes | |
class | PgemmTest< T > |
Typedefs | |
typedef ::testing::Types< double, std::complex< double > > | MyTypes |
Functions | |
void | random_data (std::vector< double > &A_global, std::vector< double > &B_global, std::vector< double > &Cref_global, std::vector< double > &C_global, double &alpha, double &beta) |
void | random_data (std::vector< std::complex< double > > &A_global, std::vector< std::complex< double > > &B_global, std::vector< std::complex< double > > &Cref_global, std::vector< std::complex< double > > &C_global, std::complex< double > &alpha, std::complex< double > &beta) |
double | get_double (std::complex< double > &val) |
double | get_double (double &val) |
void | scatterv_data (const double *sendbuf, const int *sendcounts, const int *displs, double *recvbuf, const int recvcount, MPI_Comm comm) |
void | scatterv_data (const std::complex< double > *sendbuf, const int *sendcounts, const int *displs, std::complex< double > *recvbuf, const int recvcount, MPI_Comm comm) |
TYPED_TEST_SUITE (PgemmTest, MyTypes) | |
TYPED_TEST (PgemmTest, even_case) | |
TYPED_TEST (PgemmTest, odd_case) | |
TYPED_TEST (PgemmTest, row_parallel) | |
TYPED_TEST (PgemmTest, col_parallel) | |
TYPED_TEST (PgemmTest, divide_col) | |
TYPED_TEST (PgemmTest, divide_row) | |
int | main (int argc, char **argv) |
typedef ::testing::Types<double, std::complex<double> > MyTypes |
double get_double | ( | double & | val | ) |
double get_double | ( | std::complex< double > & | val | ) |
int main | ( | int | argc, |
char ** | argv | ||
) |
void random_data | ( | std::vector< double > & | A_global, |
std::vector< double > & | B_global, | ||
std::vector< double > & | Cref_global, | ||
std::vector< double > & | C_global, | ||
double & | alpha, | ||
double & | beta | ||
) |
void random_data | ( | std::vector< std::complex< double > > & | A_global, |
std::vector< std::complex< double > > & | B_global, | ||
std::vector< std::complex< double > > & | Cref_global, | ||
std::vector< std::complex< double > > & | C_global, | ||
std::complex< double > & | alpha, | ||
std::complex< double > & | beta | ||
) |
void scatterv_data | ( | const double * | sendbuf, |
const int * | sendcounts, | ||
const int * | displs, | ||
double * | recvbuf, | ||
const int | recvcount, | ||
MPI_Comm | comm | ||
) |
void scatterv_data | ( | const std::complex< double > * | sendbuf, |
const int * | sendcounts, | ||
const int * | displs, | ||
std::complex< double > * | recvbuf, | ||
const int | recvcount, | ||
MPI_Comm | comm | ||
) |
TYPED_TEST | ( | PgemmTest | , |
col_parallel | |||
) |
TYPED_TEST | ( | PgemmTest | , |
divide_col | |||
) |
TYPED_TEST | ( | PgemmTest | , |
divide_row | |||
) |
TYPED_TEST | ( | PgemmTest | , |
even_case | |||
) |
TYPED_TEST | ( | PgemmTest | , |
odd_case | |||
) |
TYPED_TEST | ( | PgemmTest | , |
row_parallel | |||
) |