1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
#include "BLAS.h"
#include "GemmR.h"
#include "Matrix.h"
#include "Parallelizer2.h"
#include "Random48.h"
typedef double RealType;
template <typename SomeRngType>
void fillRandom(PsimagLite::Matrix<RealType>& m, RealType min, RealType max, SomeRngType& rng)
{
const SizeType rows = m.rows();
const SizeType cols = m.cols();
for (SizeType i = 0; i < rows; ++i)
for (SizeType j = 0; j < cols; ++j)
m(i, j) = min + rng() * max;
}
bool equalMatrices(PsimagLite::Matrix<RealType>& a,
PsimagLite::Matrix<RealType>& b,
RealType tolerance)
{
const SizeType rows = a.rows();
const SizeType cols = b.cols();
if (rows != b.rows() || cols != b.cols())
return false;
for (SizeType i = 0; i < rows; ++i)
for (SizeType j = 0; j < cols; ++j)
if (fabs(a(i, j) - b(i, j)) > tolerance)
return false;
return true;
}
int main(int argc, char** argv)
{
if (argc < 2)
throw PsimagLite::RuntimeError("USAGE: " + PsimagLite::String(argv[0]) + " total nthreadsOuter nthreadsInner\n");
const bool needsPrinting = false;
int const nb = 99;
int total = atoi(argv[1]);
int nthreadsOuter = atoi(argv[2]);
int nthreadsInner = atoi(argv[3]);
PsimagLite::Concurrency concurrency(&argc, &argv, nthreadsInner);
PsimagLite::Random48<RealType> rng(1234);
auto lambda = [&rng, nthreadsInner](SizeType, SizeType) {
PsimagLite::GemmR<RealType> gemmR(needsPrinting, nb, nthreadsInner);
SizeType lda = static_cast<SizeType>(rng() * 500) + 10;
SizeType cda = lda;
SizeType ldb = lda;
SizeType cdb = lda;
SizeType ldc = lda;
SizeType cdc = lda;
PsimagLite::Matrix<RealType> A(lda, cda);
PsimagLite::Matrix<RealType> B(ldb, cdb);
PsimagLite::Matrix<RealType> C(ldc, cdc);
fillRandom(A, -10, 10, rng);
fillRandom(B, -10, 10, rng);
gemmR('N', 'N', ldc, cdc, cda, 1.0, &A(0, 0), lda, &B(0, 0), ldb, 0.0, &C(0, 0), ldc);
PsimagLite::Matrix<RealType> C2(ldc, cdc);
psimag::BLAS::GEMM('N', 'N', ldc, cdc, cda, 1.0, &A(0, 0), lda, &B(0, 0), ldb, 0.0, &C2(0, 0), ldc);
if (!equalMatrices(C, C2, 1e-6))
throw PsimagLite::RuntimeError("TEST FAILED\n");
};
PsimagLite::CodeSectionParams csp = PsimagLite::Concurrency::codeSectionParams;
csp.npthreads = nthreadsOuter;
PsimagLite::Parallelizer2<> parallelizer2(csp);
parallelizer2.parallelFor(0, total, lambda);
}
|
