Revert to previous version.
authorJianzhong Xu <a0869574@ti.com>
Fri, 27 Mar 2015 18:22:13 +0000 (14:22 -0400)
committerJianzhong Xu <a0869574@ti.com>
Fri, 27 Mar 2015 18:22:13 +0000 (14:22 -0400)
examples/ludinv/main.c

index 610f8d3b54dd1ee2ae3cfa721366a8329de81926..9dbbd2b99942c1b777e6b98341aaa5a9ef6542ca 100644 (file)
@@ -34,9 +34,6 @@
 extern "C" {
 #endif
 #include "cblas.h"
-#include "f2c.h"
-#include "blaswrap.h"
-#include "clapack.h"
 #ifdef __cplusplus
 }
 #endif
@@ -49,33 +46,47 @@ struct timespec t0,t1;
 #define tock() (clock_gettime(CLOCK_MONOTONIC, &t1), \
                         t1.tv_sec - t0.tv_sec + (t1.tv_nsec - t0.tv_nsec) / 1e9)
 
+/* LAPACK routines */
+extern int dgetrf_(int *, int *, double *, int *, int *, int *);
+extern int dgetri_(int *, double *, int *, int *, double *, int *, int *);
+extern int dgeqrf_(int *, int *, double *, int *, double *, double *, int *, int *);
+extern int dtrtri_(char *, char *, int *, double *, int *, int *);
+
+extern int dlatmr_(int *, int *, char *, int *, char *, 
+                   double *, int *, double *, double *, char *, char *,
+                   double *, int *, double *, double *, int *, 
+                   double *, char *, int *, int *, int *, 
+                   double *, double *, char *, double *, int *, 
+                   int *, int *);
+extern int dlacpy_(char *, int *, int *, double *, int *, double *, int *);
+
 /* Auxiliary routines prototypes */
-void print_int_vector( char* desc, integer n, integer* a );
-void print_real_matrix(char* desc, doublereal *mat, integer n, integer m, integer ldv ); 
-doublereal compute_inverse_error(doublereal *matrix_A, doublereal *matrix_A_inverse, integer n, integer ld);
-void print_performance(integer *matrix_sizes, float *secs_runs, integer num_test, integer num_run);
+void print_int_vector( char* desc, int n, int* a );
+void print_real_matrix(char* desc, double *mat, int n, int m, int ldv ); 
+double compute_inverse_error(double *matrix_A, double *matrix_A_inverse, int n, int ld);
+void print_performance(int *matrix_sizes, float *secs_runs, int num_test, int num_run);
 
 /* Parameters */
 #define MAX_ALLOWED_ERR (1e-20)
 
 /* Main program */
-integer print_data;
-integer main(integer argc, char *argv[]) {
+int print_data;
+int main(int argc, char *argv[]) {
     /* Locals */
     float secs, secs_lud_inv, *secs_all_runs;
-    integer num_test, num_run, i, j;
-    integer n, n_min, n_inc, info, lwork, lda;
-    doublereal lwork_opt, max_abs_error;
-    doublereal *work;       
-    doublereal *matrix_A, *matrix_LUD;
-    doublereal *vec_dbr_diagr, *vec_dbr_diag, *vec_dbr_diagl;
-    integer    *vec_int_pivot, *vec_int_work;
-    integer    *matrix_sizes;
+    int num_test, num_run, i, j;
+    int n, n_min, n_inc, info, lwork, lda;
+    double lwork_opt, max_abs_error;
+    double *work;       
+    double *matrix_A, *matrix_LUD;
+    double *vec_dbr_diagr, *vec_dbr_diag, *vec_dbr_diagl;
+    int    *vec_int_pivot, *vec_int_work;
+    int    *matrix_sizes;
     
-    integer const_int6    = 6;
-    doublereal const_dbr1 = 1.;
-    doublereal const_dbr0 = 0.;
-    integer rand_seed[4] = {10,123,278,3579};  /* numbers must lie between 0 and 4095 */
+    int const_int6    = 6;
+    double const_dbr1 = 1.;
+    double const_dbr0 = 0.;
+    int rand_seed[4] = {10,123,278,3579};  /* numbers must lie between 0 and 4095 */
 
     fprintf(stdout, "\n\n===== LAPACK example: matrix inversion through LUD =====\n");
     if(argc == 1) { /* no command line arguments, use default */
@@ -99,45 +110,45 @@ integer main(integer argc, char *argv[]) {
     rand_seed[0] = 4675;
     
     secs_all_runs = (float *)malloc(sizeof(float)*num_test*num_run);
-    matrix_sizes  = (integer*)malloc(sizeof(integer)*num_test);
+    matrix_sizes  = (int*)malloc(sizeof(int)*num_test);
     
     for(i=0, n=n_min; i<num_test; i++, n+=n_inc) 
     {
         lda = n;
         matrix_sizes[i] = n;
 
-        printf("\n\nStart matrix inversion by LUD for matrix size %d. \n", (int)n);
+        printf("\n\nStart matrix inversion by LUD for matrix size %d. \n", n);
         
         /* Operation counts of LUD inversion according to LAPACK Working Note 41 */
         /*operation_counts = 2.*n*n*n - 1.5*n*n + 2.5*n;  */
         
         /* Allocate memory for matrix and vectors */
-        matrix_A      = (doublereal *)__malloc_ddr(sizeof(doublereal)*n*n);
-        matrix_LUD    = (doublereal *)__malloc_ddr(sizeof(doublereal)*n*n);
-        vec_int_pivot = (integer    *)__malloc_ddr(sizeof(integer)*n);     
-        vec_int_work  = (integer    *)__malloc_ddr(sizeof(integer)*n);     
-        vec_dbr_diag  = (doublereal *)__malloc_ddr(sizeof(doublereal)*n);  
-        vec_dbr_diagl = (doublereal *)__malloc_ddr(sizeof(doublereal)*n);  
-        vec_dbr_diagr = (doublereal *)__malloc_ddr(sizeof(doublereal)*n);  
+        matrix_A      = (double *)__malloc_ddr(sizeof(double)*n*n);
+        matrix_LUD    = (double *)__malloc_ddr(sizeof(double)*n*n);
+        vec_int_pivot = (int    *)__malloc_ddr(sizeof(int)*n);     
+        vec_int_work  = (int    *)__malloc_ddr(sizeof(int)*n);     
+        vec_dbr_diag  = (double *)__malloc_ddr(sizeof(double)*n);  
+        vec_dbr_diagl = (double *)__malloc_ddr(sizeof(double)*n);  
+        vec_dbr_diagr = (double *)__malloc_ddr(sizeof(double)*n);  
                 
         j = 0;
         while(j<num_run) {
             /* Generate a N by N random matrix */
-            printf("\nStart test run %d out of %d. \n", (int)(j+1), (int)num_run);
-            printf("Generate %d by %d random matrix. \n", (int)n, (int)n);
+            printf("\nStart test run %d out of %d. \n", j+1, num_run);
+            printf("Generate %d by %d random matrix. \n", n, n);
             dlatmr_(&n, &n, "S", rand_seed, "N", vec_dbr_diag, &const_int6, &const_dbr1, 
                     &const_dbr1, "T", "N", vec_dbr_diagl, &const_int6, &const_dbr1, vec_dbr_diagr, 
                     &const_int6, &const_dbr1, "N", vec_int_pivot, &n, &n, &const_dbr0,
                     &const_dbr1, "N", matrix_A, &lda, vec_int_work, &info);
             if(info != 0) {
-                printf("Random matrix generation has a problem. Returned error value is %d. Start a new run. \n", (int)info);
+                printf("Random matrix generation has a problem. Returned error value is %d. Start a new run. \n", info);
                 //goto START_NEW_RUN;
                 exit(-1);
             }
         
-            memcpy((void *)matrix_LUD, (void *)matrix_A, sizeof(doublereal)*n*n);
+            memcpy((void *)matrix_LUD, (void *)matrix_A, sizeof(double)*n*n);
             
-            print_real_matrix("Generated random matrix (doublereal real): ", matrix_LUD, n, n, lda);
+            print_real_matrix("Generated random matrix (double real): ", matrix_LUD, n, n, lda);
 
             tick();
             
@@ -147,7 +158,7 @@ integer main(integer argc, char *argv[]) {
             secs_lud_inv = tock();
             
             if(info != 0) {
-                printf("LU factorization has a problem. Returned error value of dgetrf is %d. Start a new run. \n", (int)info);
+                printf("LU factorization has a problem. Returned error value of dgetrf is %d. Start a new run. \n", info);
                 //goto START_NEW_RUN;
                 exit(-1);
             }
@@ -165,8 +176,8 @@ integer main(integer argc, char *argv[]) {
             /* Query and allocate the optimal workspace */
             lwork = -1;
             dgetri_(&n, matrix_LUD, &lda, vec_int_pivot, &lwork_opt, &lwork, &info);        
-            lwork = (integer)lwork_opt;
-            work = (doublereal *)__malloc_ddr( lwork*sizeof(doublereal) );
+            lwork = (int)lwork_opt;
+            work = (double *)__malloc_ddr( lwork*sizeof(double) );
             
             /* Compute inversion of matrix based on LU factorization */
             dgetri_(&n, matrix_LUD, &lda, vec_int_pivot, work, &lwork, &info);    
@@ -176,7 +187,7 @@ integer main(integer argc, char *argv[]) {
             __free_ddr((void *)work);
             if(info != 0) {
                 printf("Matrix inversion through LU factorization has a problem.");
-                printf("Returned error value of dgetri is %d. Start a new run. \n", (int)info);
+                printf("Returned error value of dgetri is %d. Start a new run. \n", info);
                 //goto START_NEW_RUN;
                 exit(-1);
             }
@@ -222,22 +233,22 @@ START_NEW_RUN:
 
 
 /* Auxiliary routine: printing a vector of integers */
-void print_int_vector( char* desc, integer n, integer* a ) 
+void print_int_vector( char* desc, int n, int* a ) 
 {
-   integer j;
+   int j;
 
    if (!print_data)
      return;
         
    printf( "\n %s\n", desc );
-   for( j = 0; j < n; j++ ) printf( " %6i", (int)a[j] );
+   for( j = 0; j < n; j++ ) printf( " %6i", a[j] );
    printf( "\n" );
 }
 
 /* Auxiliary routine: printing a real matrix */
-void print_real_matrix(char* desc, doublereal *mat, integer m, integer n, integer ldv ) 
+void print_real_matrix(char* desc, double *mat, int m, int n, int ldv ) 
 {
-   integer i, j;
+   int i, j;
 
    if (!print_data)
      return;
@@ -252,14 +263,14 @@ void print_real_matrix(char* desc, doublereal *mat, integer m, integer n, intege
 }
 
 
-doublereal compute_inverse_error(doublereal *matrix_A, doublereal *matrix_A_inverse, integer n, integer ld)
+double compute_inverse_error(double *matrix_A, double *matrix_A_inverse, int n, int ld)
 {
-   integer i, j;
-   doublereal abs_err, relative_err;
-   doublereal *matrix_A_mult_A_inv;
-   doublereal max_err = 0.;
+   int i, j;
+   double abs_err, relative_err;
+   double *matrix_A_mult_A_inv;
+   double max_err = 0.;
 
-   matrix_A_mult_A_inv = (doublereal *)malloc(n*n*sizeof(doublereal));
+   matrix_A_mult_A_inv = (double *)malloc(n*n*sizeof(double));
    
    cblas_dgemm(CblasColMajor,CblasNoTrans,CblasNoTrans,n,n,n,1.0,matrix_A,ld,
                matrix_A_inverse,ld,0.0,matrix_A_mult_A_inv,ld);
@@ -287,9 +298,9 @@ doublereal compute_inverse_error(doublereal *matrix_A, doublereal *matrix_A_inve
 }
 
 
-void print_performance(integer *matrix_sizes, float *secs_runs, integer num_test, integer num_run)
+void print_performance(int *matrix_sizes, float *secs_runs, int num_test, int num_run)
 {
-    integer i, j;
+    int i, j;
     float inst, min, max;
     float tot, avg;
     
@@ -300,19 +311,19 @@ void print_performance(integer *matrix_sizes, float *secs_runs, integer num_test
     
     for(i=0; i<num_test; i++)
     {
-        printf("%11d ", (int)matrix_sizes[i]);
+        printf("%11d ", matrix_sizes[i]);
         tot = 0.;
         max = 0.;
         min = 1e100;
         for(j=0; j<num_run; j++) 
           {
             inst = secs_runs[i*num_run+j];            
-            tot += (doublereal)inst;
+            tot += (double)inst;
             
             if(inst>max) max = inst;
             if(inst<min) min = inst;
           }
-        avg = tot/(doublereal)num_run;
+        avg = tot/(double)num_run;
         printf("%12f%12f%12f\n", min, max, avg);
     }
 }