[991] | 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 | }
|
---|