|
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 | |||
| ) |