1 | #include <stdlib.h>
|
---|
2 | #include <stdio.h>
|
---|
3 |
|
---|
4 | void InitTim();
|
---|
5 | void PrtTim(char *);
|
---|
6 |
|
---|
7 | void Mult(int n, double *v1, double *v2, double *v3);
|
---|
8 | void MultOMP(int n, double *v1, double *v2, double *v3);
|
---|
9 |
|
---|
10 | int main (int narg, char *arg[])
|
---|
11 | {
|
---|
12 | int i,j, N, M;
|
---|
13 |
|
---|
14 | double *v1, *v2, *v3;
|
---|
15 |
|
---|
16 | if (narg < 2) {
|
---|
17 | printf("\n Usage tompC P/x [N M] \n \n");
|
---|
18 | printf(" P --> Calling MultOMP, x ---> Calling Mult \n");
|
---|
19 | printf(" N (=100): External loop number M : Array Size (500000) \n");
|
---|
20 | exit(0);
|
---|
21 | }
|
---|
22 | N = 100;
|
---|
23 | M = 500000;
|
---|
24 | if (narg > 2) N = atoi(arg[2]);
|
---|
25 | if (narg > 3) M = atoi(arg[3]);
|
---|
26 |
|
---|
27 | InitTim();
|
---|
28 |
|
---|
29 | v1 = (double *)malloc(M*sizeof(double));
|
---|
30 | v2 = (double *)malloc(M*sizeof(double));
|
---|
31 | v3 = (double *)malloc(M*sizeof(double));
|
---|
32 | for(i=0; i<M; i++) { v1[i] = random()%1000; v2[i] = random()%5000; }
|
---|
33 | PrtTim("End of Init ");
|
---|
34 |
|
---|
35 | if (*arg[1] == 'P') {
|
---|
36 | printf("Calling N=%d times MultOMP(Size= %d) \n", N, M);
|
---|
37 | for(i=0; i<N; i++) MultOMP(M, v1, v2, v3);
|
---|
38 | }
|
---|
39 | else {
|
---|
40 | printf("Calling N=%d times Mult(Size= %d) \n", N, M);
|
---|
41 | for(i=0; i<N; i++) Mult(M, v1, v2, v3);
|
---|
42 | }
|
---|
43 |
|
---|
44 | PrtTim("End of Multiplication ");
|
---|
45 |
|
---|
46 | free(v1);
|
---|
47 | free(v2);
|
---|
48 | free(v3);
|
---|
49 |
|
---|
50 | exit(0);
|
---|
51 | }
|
---|
52 |
|
---|
53 |
|
---|
54 | void Mult(int n, double *v1, double *v2, double *v3)
|
---|
55 | {
|
---|
56 | int k;
|
---|
57 | for(k=0; k<n; k++)
|
---|
58 | v3[k] = v1[k] * v2[k];
|
---|
59 | }
|
---|
60 |
|
---|
61 | void MultOMP(int n, double *v1, double *v2, double *v3)
|
---|
62 | {
|
---|
63 | int k;
|
---|
64 | #pragma omp parallel for
|
---|
65 | for(k=0; k<n; k++)
|
---|
66 | v3[k] = v1[k] * v2[k];
|
---|
67 | }
|
---|