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

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

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1805 - (hide annotations)
Wed Sep 24 23:39:00 2008 UTC (10 years, 8 months ago) by ksteube
File size: 2469 byte(s)
Added python-level methods getMPISizeWorld() and getMPIRankWorld() for MPI process info.
Test suite run_inputOutput.py runs on any number of cores now, hybrid may still be a problem.

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 phornby 1628 #ifdef _WIN32
28     #include <WinSock2.h>
29     #endif
30    
31 gross 391 namespace escript {
32    
33 ksteube 1247 int getSvnVersion()
34     {
35     #ifdef SVN_VERSION
36     return SVN_VERSION;
37     #else
38     return 0;
39     #endif
40     }
41    
42 ksteube 1620 /* 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 ksteube 1561 void printParallelThreadCnt()
68     {
69     int mpi_iam=0, mpi_num=1;
70 ksteube 1568 char hname[64];
71 ksteube 1561
72 ksteube 1705 #ifdef HAVE_GETHOSTNAME
73 ksteube 1568 gethostname(hname, 64);
74 ksteube 1705 #else
75     strcpy(hname, "unknown host");
76     #endif
77 ksteube 1567
78 ksteube 1561 #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     #pragma omp parallel
84     {
85     int omp_iam=0, omp_num=1;
86     #ifdef _OPENMP
87     omp_iam = omp_get_thread_num(); /* Call in a parallel region */
88     omp_num = omp_get_num_threads();
89     #endif
90 ksteube 1620 printf("printParallelThreadCounts: MPI=%03d/%03d OpenMP=%03d/%03d running on %s core %d\n",
91     mpi_iam, mpi_num, omp_iam, omp_num, hname, get_core_id());
92 ksteube 1561 }
93     }
94    
95 gross 391 void setNumberOfThreads(const int num_threads)
96     {
97    
98     #ifdef _OPENMP
99     omp_set_num_threads(num_threads);
100     #endif
101    
102     }
103    
104     int getNumberOfThreads()
105     {
106     #ifdef _OPENMP
107     return omp_get_max_threads();
108     #else
109     return 1;
110     #endif
111    
112     }
113    
114 ksteube 1805 int getMPISizeWorld() {
115     int mpi_num = 1;
116     #ifdef PASO_MPI
117     MPI_Comm_size(MPI_COMM_WORLD, &mpi_num);
118     #endif
119     return mpi_num;
120     }
121    
122     int getMPIRankWorld() {
123     int mpi_iam = 0;
124     #ifdef PASO_MPI
125     MPI_Comm_rank(MPI_COMM_WORLD, &mpi_iam);
126     #endif
127     return mpi_iam;
128     }
129    
130 gross 391 } // end of namespace

  ViewVC Help
Powered by ViewVC 1.1.26