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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
#include <mpi.h>
#include <sys/time.h>
#include <iostream>
#include <functional>
#include <algorithm>
#include <vector>
#include <string>
#include <sstream>
#include <stdint.h>
#include "CombBLAS/CombBLAS.h"
using namespace combblas;
double cblas_alltoalltime;
double cblas_allgathertime;
typedef SelectMaxSRing<bool, int64_t> SR;
typedef SpParMat < int64_t, bool, SpDCCols<int64_t,bool> > PSpMat_Bool;
typedef SpParMat < int64_t, int, SpDCCols<int64_t,int> > PSpMat_Int;
#define ValueType int64_t
typedef SpDCCols<int64_t,ValueType> DCColsType;
typedef SpParMat < int64_t, ValueType, DCColsType > PSpMat_Int64;
typedef PSpMat_Int64 MatType;
DECLARE_PROMOTE(MatType, MatType, MatType)
DECLARE_PROMOTE(DCColsType, DCColsType, DCColsType)
int main(int argc, char* argv[])
{
int torusi[] = {0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15};
int torusj[] = {3,0,1,2,7,4,5,6,11,8,9,10,15,12,13,14,1,2,3,0,5,6,7,4,9,10,11,8,13,14,15,12,12,13,14,15,0,1,2,3,4,5,6,7,8,9,10,11,4,5,6,7,8,9,10,11,12,13,14,15,0,1,2,3};
int nprocs, myrank;
MPI_Init(&argc, &argv);
MPI_Comm_size(MPI_COMM_WORLD,&nprocs);
MPI_Comm_rank(MPI_COMM_WORLD,&myrank);
{
SpParHelper::Print("Usage: SpMMError [arg]\nIf arg is present then matrix will be read from torus.mtx (no error).\nIf arg is absent, matrix will be created from vectors (error present)\n\n");
// Declare objects
MatType G1;
MatType G2;
if (argc > 1)
{
SpParHelper::Print("Reading torus.mtx\n");
std::ifstream input("torus.mtx");
G1.ReadDistribute(input, 0); // read it from file
ifstream input2("torus.mtx");
G2.ReadDistribute(input2, 0);
SpParHelper::Print("Read input\n");
}
else
{
SpParHelper::Print("Creating matrices from built-in vectors\n");
FullyDistVec<int64_t, ValueType> dpvi(64, 0, 0);
FullyDistVec<int64_t, ValueType> dpvj(64, 0, 0);
FullyDistVec<int64_t, ValueType> dpvv(64, 1, 0);
for (int i = 0; i <64; i++)
{
dpvi.SetElement(i, torusi[i]);
dpvj.SetElement(i, torusj[i]);
}
dpvi.DebugPrint();
dpvj.DebugPrint();
G1 = MatType(16, 16, dpvi, dpvj, dpvv);
G2 = MatType(16, 16, dpvi, dpvj, dpvv);
ofstream out1("G1.txt");
ofstream out2("G2.txt");
out1 << G1;
out2 << G2;
}
MatType G3(G1);
G1.PrintInfo();
G2.PrintInfo();
G3.PrintInfo();
SpParHelper::Print("The nnz values should be 112, 112, 112:\n");
MatType G12 = Mult_AnXBn_Synch<PlusTimesSRing<int64_t, ValueType> >(G1, G2);
G12.PrintInfo();
ofstream out12("G12.txt");
out12 << G12;
MatType G13 = Mult_AnXBn_Synch<PlusTimesSRing<int64_t, ValueType> >(G1, G3);
G13.PrintInfo();
MatType G23 = Mult_AnXBn_Synch<PlusTimesSRing<int64_t, ValueType> >(G2, G3);
G23.PrintInfo();
}
MPI_Finalize();
return 0;
}
|
