escript/src/Utils.cpp


/* $Id$ */

/*******************************************************
 *
 *           Copyright 2003-2007 by ACceSS MNRF
 *       Copyright 2007 by University of Queensland
 *
 *                http://esscc.uq.edu.au
 *        Primary Business: Queensland, Australia
 *  Licensed under the Open Software License version 3.0
 *     http://www.opensource.org/licenses/osl-3.0.php
 *
 *******************************************************/

#include "Utils.h"
#include "DataVector.h"

#ifdef _OPENMP
#include <omp.h>
#endif

#ifdef PASO_MPI
#include <mpi.h>
#endif

namespace escript {

int getSvnVersion() 
{
#ifdef SVN_VERSION
  return SVN_VERSION;
#else
  return 0;
#endif
}

void printParallelThreadCnt() 
{
  int mpi_iam=0, mpi_num=1;

  #ifdef PASO_MPI
  MPI_Comm_rank(MPI_COMM_WORLD, &mpi_iam);
  MPI_Comm_size(MPI_COMM_WORLD, &mpi_num);
  #endif

  #ifdef _OPENMP
  #pragma omp parallel
  #endif
  {
    int omp_iam=0, omp_num=1;
    #ifdef _OPENMP
    omp_iam = omp_get_thread_num(); /* Call in a parallel region */
    omp_num = omp_get_num_threads();
    #endif
    printf("printParallelThreadCounts: OpenMP=%d/%d MPI=%d/%d\n", omp_iam, omp_num, mpi_iam, mpi_num);
  }
}

void setNumberOfThreads(const int num_threads) 
{

   #ifdef _OPENMP
   omp_set_num_threads(num_threads);
   #endif

}

int getNumberOfThreads() 
{
   #ifdef _OPENMP
   return omp_get_max_threads();
   #else
   return 1;
   #endif

}

}  // end of namespace
1	gross	391
2	ksteube	1312	/* $Id$ */
3
4			/*******************************************************
5			*
6			* Copyright 2003-2007 by ACceSS MNRF
7			* Copyright 2007 by University of Queensland
8			*
9			* http://esscc.uq.edu.au
10			* Primary Business: Queensland, Australia
11			* Licensed under the Open Software License version 3.0
12			* http://www.opensource.org/licenses/osl-3.0.php
13			*
14			*******************************************************/
15
16	jgs	474	#include "Utils.h"
17	gross	797	#include "DataVector.h"
18	gross	391
19	jgs	478	#ifdef _OPENMP
20			#include <omp.h>
21			#endif
22
23	ksteube	1561	#ifdef PASO_MPI
24			#include <mpi.h>
25			#endif
26
27	gross	391	namespace escript {
28
29	ksteube	1247	int getSvnVersion()
30			{
31			#ifdef SVN_VERSION
32			return SVN_VERSION;
33			#else
34			return 0;
35			#endif
36			}
37
38	ksteube	1561	void printParallelThreadCnt()
39			{
40			int mpi_iam=0, mpi_num=1;
41
42			#ifdef PASO_MPI
43			MPI_Comm_rank(MPI_COMM_WORLD, &mpi_iam);
44			MPI_Comm_size(MPI_COMM_WORLD, &mpi_num);
45			#endif
46
47			#ifdef _OPENMP
48			#pragma omp parallel
49			#endif
50			{
51			int omp_iam=0, omp_num=1;
52			#ifdef _OPENMP
53			omp_iam = omp_get_thread_num(); /* Call in a parallel region */
54			omp_num = omp_get_num_threads();
55			#endif
56			printf("printParallelThreadCounts: OpenMP=%d/%d MPI=%d/%d\n", omp_iam, omp_num, mpi_iam, mpi_num);
57			}
58			}
59
60	gross	391	void setNumberOfThreads(const int num_threads)
61			{
62
63			#ifdef _OPENMP
64			omp_set_num_threads(num_threads);
65			#endif
66
67			}
68
69			int getNumberOfThreads()
70			{
71			#ifdef _OPENMP
72			return omp_get_max_threads();
73			#else
74			return 1;
75			#endif
76
77			}
78
79			} // end of namespace