/[escript]/trunk/escript/src/Utils.cpp
ViewVC logotype

Contents of /trunk/escript/src/Utils.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1705 - (show annotations)
Thu Aug 14 05:56:40 2008 UTC (11 years ago) by ksteube
File size: 2227 byte(s)
Branch scons-dev is hereby closed.
Some parts of scons scripts have been re-written.

1
2 /* $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 #include "Utils.h"
17 #include "DataVector.h"
18
19 #ifdef _OPENMP
20 #include <omp.h>
21 #endif
22
23 #ifdef PASO_MPI
24 #include <mpi.h>
25 #endif
26
27 #ifdef _WIN32
28 #include <WinSock2.h>
29 #endif
30
31 namespace escript {
32
33 int getSvnVersion()
34 {
35 #ifdef SVN_VERSION
36 return SVN_VERSION;
37 #else
38 return 0;
39 #endif
40 }
41
42 /* This is probably not very robust, but it works on Savanna today and is useful for performance analysis */
43 int get_core_id() {
44 int processor_num=-1;
45 #ifdef CORE_ID1
46 FILE *fp;
47 int i, count_spaces=0;
48 char fname[100];
49 char buf[1000];
50
51 sprintf(fname, "/proc/%d/stat", getpid());
52 fp = fopen(fname, "r");
53 if (fp == NULL) return(-1);
54 fgets(buf, 1000, fp);
55 fclose(fp);
56
57 for (i=strlen(buf)-1; i>=0; i--) {
58 if (buf[i] == ' ') count_spaces++;
59 if (count_spaces == 4) break;
60 }
61 processor_num = atoi(&buf[i+1]);
62 #endif
63 return(processor_num);
64 }
65
66
67 void printParallelThreadCnt()
68 {
69 int mpi_iam=0, mpi_num=1;
70 char hname[64];
71
72 #ifdef HAVE_GETHOSTNAME
73 gethostname(hname, 64);
74 #else
75 strcpy(hname, "unknown host");
76 #endif
77
78 #ifdef PASO_MPI
79 MPI_Comm_rank(MPI_COMM_WORLD, &mpi_iam);
80 MPI_Comm_size(MPI_COMM_WORLD, &mpi_num);
81 #endif
82
83 #ifdef _OPENMP
84 #pragma omp parallel
85 #endif
86 {
87 int omp_iam=0, omp_num=1;
88 #ifdef _OPENMP
89 omp_iam = omp_get_thread_num(); /* Call in a parallel region */
90 omp_num = omp_get_num_threads();
91 #endif
92 printf("printParallelThreadCounts: MPI=%03d/%03d OpenMP=%03d/%03d running on %s core %d\n",
93 mpi_iam, mpi_num, omp_iam, omp_num, hname, get_core_id());
94 }
95 }
96
97 void setNumberOfThreads(const int num_threads)
98 {
99
100 #ifdef _OPENMP
101 omp_set_num_threads(num_threads);
102 #endif
103
104 }
105
106 int getNumberOfThreads()
107 {
108 #ifdef _OPENMP
109 return omp_get_max_threads();
110 #else
111 return 1;
112 #endif
113
114 }
115
116 } // end of namespace

  ViewVC Help
Powered by ViewVC 1.1.26