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

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

Parent Directory Parent Directory | Revision Log Revision Log


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

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     #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 ksteube 1620 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 ksteube 1561 }
95     }
96    
97 gross 391 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