Difference between revisions of "Profiling GEOS-Chem with the TAU performance system"

From Geos-chem
Jump to: navigation, search
 
(6 intermediate revisions by one other user not shown)
Line 1: Line 1:
__FORCETOC__
 
'''''[[GEOS-Chem 1-month benchmark timing results|Previous]] | [[Speeding up GEOS-Chem|Next]] | [[Guide to GEOS-Chem performance]]'''''
 
 
#[[Parallelizing GEOS-Chem]]
 
#[[GEOS-Chem 7-day timing tests]]
 
#[[GEOS-Chem scalability]]
 
#[[GEOS-Chem 1-month benchmark timing results]]
 
#<span style="color:blue">'''Profiling GEOS-Chem with the TAU performance system'''</span>
 
#[[Speeding up GEOS-Chem]]
 
 
 
<span style="color:red">'''''NOTE: This page is still under development.  The Spack installation guide is still being validated.'''''</span>
 
 
 
== Overview ==
 
== Overview ==
  
Line 18: Line 5:
 
== Installing TAU ==
 
== Installing TAU ==
  
'''''NOTE: These instructions are still in flux.'''''
+
The best way to build TAU is with Spack.
  
The best way to install TAU is to use Spack.
+
== Compiling and running GEOS-Chem with TAU ==
  
(1) '''Download Spack''' as described [[Use_Spack_to_install_netCDF_on_your_system#Download_Spack|'''HERE''']].
+
You can use the <tt>tau_exec</tt> command to profile GEOS-Chem in order to reveal computational bottlenecks. Below is a job script that you can use. 
  
(2) '''Define environment variables''' for Spack as described [[Use_Spack_to_install_netCDF_on_your_system#Set_up_the_environment_for_Spack|'''HERE''']].
+
NOTE: The <tt>gcc.env</tt> file is the [https://github.com/geoschem/geos-chem/issues/637#issuecomment-788968697 script that loads your Spack modules].
  
(3) '''Check if Spack can find your compiler''' as described [[Use_Spack_to_install_netCDF_on_your_system#Make sure that Spack can find your compiler|'''HERE''']].
+
#!/bin/bash
 
+
(4) '''Install TAU''' with the following Spack command:
+
  #SBATCH -c 24
 
+
  #SBATCH -n 1
  spack install tau@2.28.1 openmp=True
+
  #SBATCH -p YOUR-QUEUE-NAME
 
+
  #SBATCH --mem=15000
You can also force installation with a <span style="color:blue">specific compiler version</span> by typing:
+
 
+
  spack install tau@2.28.1 <span style="color:blue">%gcc@8.2.0</span> openmp=True
+
 
+
(5) '''Define environment variables for TAU''' as shown below.  
+
 
+
  # TAU home location
+
export TAU_HOME=$(spack location -i tau)
+
 
   
 
   
  # Put the location of tau_f90.sh in the search path
+
  ### geoschem.profile.run
  export PATH=$PATH:$TAU_HOME/bin
+
  ### Job script to run GEOS-Chem with TAU profiling
 +
 
 +
source ~/.bashrc  # Load bash settings
 +
source ~/gcc.env  # Environment file that loads TAU module from Spack
 
   
 
   
  # Define options for TAU
+
  # Set the # of OpenMP threads to the # of cores we requested
  export TAU_OPTIONS="-optRevert -optVerbose -optPreProcess -optContinueBeforeOMP -optPdtGnuFortranParser"
+
  export OMP_NUM_THREADS=24
 
   
 
   
  # Define location of TAU makefile
+
  # Run GEOS-Chem Classic and have TAU profile it
  export TAU_MAKEFILE=$TAU_HOME/x86_64/lib/Makefile.tau-papi-pthread-pdt-openmp
+
  srun -c $OMP_NUM_THREADS tau_exec -T serial,openmp -ebs -ebs_resolution=function ./gcclassic > log 2>&1
  
You can place these into one of your system environment files so that they are executed on startup.
+
If you use SLURM, you can submit this with
  
== Compiling and running GEOS-Chem with TAU ==
+
sbatch geoschem.profile.run
  
To profile GEOS-Chem with TAU, you must first compile with the <tt>TAU_PROF=y</tt> Makefile option, e.g.:
+
Otherwise you can just run it interactively:
  
  # Remove files from a previous compilation with TAU (if necessary)
+
  ./geoschem.profile.run
  make tauclean
+
   
+
  # Compile with TAU profiling
+
  make TAU_PROF=y ...etc. other makefile options ...
+
 
+
It is important that you compile on a single processor (i.e. don't pass <tt>-j4</tt> or <tt>-j8</tt>) to allow TAU to properly instrument the code.
+
 
+
The <tt>TAU_PROF=y</tt> option will set <tt>COMPILE_CMD :=tau_f90.sh</tt> instead of <tt>COMPILE_CMD :=$(FC)</tt> where <tt>FC</tt> is <tt>gfortran</tt> or <tt>ifort</tt>.
+
 
+
Once you have compiled GEOS-Chem with <tt>TAU_PROF=y</tt>, you can run GEOS-Chem as you normally would. GEOS-Chem will create <tt>profile.*</tt> files containing the profiling information.
+
  
 
== Using ParaProf to create plots from the profiling data ==
 
== Using ParaProf to create plots from the profiling data ==
Line 87: Line 59:
  
 
To save the plot, select "Save as Bitmap Image" from the File menu. In the Save Image File window, you may select the output type (JPEG File or PNG file) and specify the file name and location.
 
To save the plot, select "Save as Bitmap Image" from the File menu. In the Save Image File window, you may select the output type (JPEG File or PNG file) and specify the file name and location.
 
--[[User:Melissa Payer|Melissa Sulprizio]] ([[User talk:Melissa Payer|talk]]) 22:48, 8 February 2017 (UTC)<br>--[[User:Bmy|Bob Yantosca]] ([[User talk:Bmy|talk]]) 16:55, 10 February 2017 (UTC)
 
 
 
----
 
'''''[[GEOS-Chem 1-month benchmark timing results|Previous]] | [[Speeding up GEOS-Chem|Next]] | [[Guide to GEOS-Chem performance]]'''''
 

Latest revision as of 18:12, 24 October 2023

Overview

The TAU Performance System is a profiling tool for performance analysis of parallel programs in Fortran, C, C++, Java, and Python. TAU uses a visualization tool, ParaProf, to create graphical displays of the performance analysis results.

Installing TAU

The best way to build TAU is with Spack.

Compiling and running GEOS-Chem with TAU

You can use the tau_exec command to profile GEOS-Chem in order to reveal computational bottlenecks. Below is a job script that you can use.

NOTE: The gcc.env file is the script that loads your Spack modules.

#!/bin/bash

#SBATCH -c 24
#SBATCH -n 1
#SBATCH -p YOUR-QUEUE-NAME
#SBATCH --mem=15000

### geoschem.profile.run
### Job script to run GEOS-Chem with TAU profiling
 
source ~/.bashrc  # Load bash settings
source ~/gcc.env  # Environment file that loads TAU module from Spack

# Set the # of OpenMP threads to the # of cores we requested
export OMP_NUM_THREADS=24

# Run GEOS-Chem Classic and have TAU profile it
srun -c $OMP_NUM_THREADS tau_exec -T serial,openmp -ebs -ebs_resolution=function ./gcclassic > log 2>&1

If you use SLURM, you can submit this with

sbatch geoschem.profile.run

Otherwise you can just run it interactively:

./geoschem.profile.run

Using ParaProf to create plots from the profiling data

In your run directory, there should be one or more profile.* files. The number of profile.* files will depend on the number of CPUs that you use for your GEOS-Chem simulation. To pack all of the profiling data into a single file, type:

paraprof --pack GEOS-Chem_Profile_Results.ppk

Then run paraprof on the packed format (.ppk) file using:

paraprof GEOS-Chem_Profile_Results.ppk

This will open two windows, the ParaProf Manager window and the Main Data window. For more information on how to interpret the profiling data, see the ParaProf User's Manual.

If you click on the the bar labeled "thread0" in the ParaProf manager window, you can generate a plot that looks like this:

V11-01-thread0.png

The value displayed, the units, and the sort order can be changed from the Options menu. The time that each subroutine spent on the master thread is displayed as a histogram. By examining the histogram you can see which routines are taking the longest to execute. For example, the above plot shows that the COMPUTE_F routine (highlighted with the red box) is spending 444 seconds on the master thread, which is longer than the Rosenbrock solver takes to run. This is a computational bottleneck, which was ultimately caused by an unparallelized DO loop.

To save the plot, select "Save as Bitmap Image" from the File menu. In the Save Image File window, you may select the output type (JPEG File or PNG file) and specify the file name and location.