int main(int argc, char **argv) {
int heap=300000, stack=300000;
int me, nprocs;
/* Step1: Initialize Message Passing library */
#ifdef MPI
MPI_Init(&argc, &argv); /* initialize MPI */
#else
PBEGIN_(argc, argv); /* initialize TCGMSG */
#endif
/* Step2: Initialize GA */
/* ### intialize the GA library */
GA_Initialize();
/* Step3: Initialize Memory Allocator (MA) */
if(! MA_init(C_DBL, stack, heap) ) GA_Error("MA_init failed",stack+heap);
/* ### assign the local processor ID to the int variable "me"
* ### and the total number of processors to the int variable
* ### "nprocs" */
me = GA_Nodeid();
nprocs = GA_Nnodes();
if(me==0) {
printf("\nUsing %d processes\n\n", nprocs); fflush(stdout);
}
TRANSPOSE1D();
if(me==0)printf("\nTerminating ..\n");
/* ### terminate the GA library */
GA_Terminate();
#ifdef MPI
MPI_Finalize();
#else
PEND_();
#endif
}
void tcg_pend()
{
PEND_();
}
int main(int argc, char **argv) {
int i, matrix_size;
int heap=20000000, stack=200000000;
double *A=NULL;
if(argc != 2)
{
printf("Usage Error\n\t Usage: <program> <matrix_size>\n");
exit(0);
}
matrix_size = atoi(argv[1]);
if(matrix_size <= 0)
{
printf("Error: matrix size (%d) should be > 0\n",
matrix_size);
GA_Error("matrix size should be >0", 1);
}
/* *****************************************************************
* Initialize MPI/TCGMSG-MPI, GA and MA
* *****************************************************************/
#ifdef MPI
#ifdef DCMF
int desired = MPI_THREAD_MULTIPLE;
int provided;
MPI_Init_thread(&argc, &argv, desired, &provided);
if ( provided != MPI_THREAD_MULTIPLE ) printf("provided != MPI_THREAD_MULTIPLE\n");
#else
MPI_Init (&argc, &argv); /* initialize MPI */
#endif
#else
PBEGIN_(argc, argv); /* initialize TCGMSG-MPI */
#endif
GA_Initialize(); /* initialize GA */
me = GA_Nodeid();
nprocs = GA_Nnodes();
heap /= nprocs;
stack /= nprocs;
if(! MA_init(MT_F_DBL, stack, heap)) /* initialize MA */
{
GA_Error("MA_init failed",stack+heap);
}
/* create/initialize the matrix */
if((A = (double*)malloc(matrix_size*matrix_size*sizeof(double))) == NULL)
{
GA_Error("malloc failed", matrix_size*matrix_size*sizeof(double));
}
for(i=0; i<2; i++) /* 5 runs */
{
init_array(A, matrix_size);
#if DEBUG
if(me==0) print_array(A, matrix_size);
#endif
/* *****************************************************************
* Perform LU Factorization
* *****************************************************************/
ga_lu(A, matrix_size);
}
free(A);
/* *****************************************************************
* Terminate MPI/TCGMSG-MPI, GA and MA
* *****************************************************************/
if(me==0)printf("Terminating ..\n");
GA_Terminate();
#ifdef MPI
MPI_Finalize();
#else
PEND_();
#endif
return 0;
}
void CleanUp(ArgStruct *p)
{
PEND_();
}