Kokkos Tools provide a set of light-weight of profiling and debugging utilities, which interface with instrumentation hooks built directly into the Kokkos runtime. Compared to 3rd party tools these tools can provide much cleaner, context-specific information: in particular, they allow kernel-centric analysis and they use labels provided to Kokkos constructs (kernel launches and views).
Under most circumstances, the profiling hooks are compiled into Kokkos executables by default assuming that the profiling hooks' version is compatible with the tools' version. No recompilation or changes to your build procedures are required.
Note: Kokkos
must be configured with Kokkos_ENABLE_LIBDL=ON
to load profiling hooks dynamically. This is the default for most cases anyway.
To use one of the tools you have to compile it, which will generate a dynamic library. Before executing the Kokkos application you then have to set the environment variable KOKKOS_TOOLS_LIBS
to point to the dynamic library.
CMake and Makefiles are supported for building Kokkos Tools. The following provides instructions for both.
- Create a build directory in Kokkos Tools, e.g., type
mkdir myBuild; cd myBuild
- To configure, type
cmake .. -DCMAKE_INSTALL_PREFIX=${YOUR_KOKKOS_TOOLS_INSTALL_DIR}
. There are more options but in most cases the defaults are sufficient. - To compile, type
make
- To install, type
make install
Given your installed tool shared library lib<name_of_tool_shared_lib>.so
and an application executable called yourApplication.exe, type:
export KOKKOS_TOOLS_LIBS=${YOUR_KOKKOS_TOOLS_INSTALL_DIR}/lib<name_of_tool_shared_lib>.so; ./yourApplication.exe
To build some library <name_of_tool_shared_lib>
with make, simply type make
within that library's subdirectory ${YOUR_KOKKOS_TOOLS_LIB_SRC_DIR}
of Kokkos Tools. This generates the shared library within that subdirectory.
Given your installed tool shared library <name_of_tool_shared_lib>.so
and an application executable called yourApplication.exe
, type:
export KOKKOS_TOOLS_LIBS=${YOUR_KOKKOS_TOOLS_LIB_SRC_DIR}/<name_of_tool_shared_lib>.so; ./yourApplication.exe
One can explicitly add instrumentation to a library or an application. Currently, the only hooks intended for explicit programmer use are the Region related hooks. These use a push/pop model to mark coarser regions in your code.
void foo() {
Kokkos::Profiling::pushRegion("foo");
bar();
stool();
Kokkos::Profiling::popRegion();
}
The following provides an overview of the tools available in the set of Kokkos Tools. Click on each Kokkos Tools name to see more details about the tool via the Kokkos Tools Wiki.
-
A tool which is used in conjunction with analysis tools, to restrict them to a subset of the application.
-
A tool to be used in conjunction with analysis tools to restrict the tooling to samples of Kokkos kernel invocations.
-
This tool outputs the high water mark of memory usage of the application. The high water mark of memory usage is the highest amount of memory that is being utilized during the application's execution.
-
Generates a per Memory Space timeline of memory utilization.
-
Tool to track memory events such as allocation and deallocation. It also provides the information of the MemoryUsage tool.
-
Captures basic timing information for Kernels.
-
Prints Kokkos Kernel and Region events during runtime.
-
Provides Kokkos Kernel Names to VTune, so that analysis can be performed on a per kernel base.
-
Like VTuneConnector but turns profiling off outside of kernels. Should be used in conjunction with the KernelFilter tool.
-
Provides Kokkos Kernel Names to NVTX, so that analysis can be performed on a per kernel base.
-
Modular connector for accumulating timing, memory usage, hardware counters, and other various metrics. Supports controlling VTune, CUDA profilers, and TAU + kernel name forwarding to VTune, NVTX, TAU, Caliper, and LIKWID.
Defining a timemory component will enable your plug-in to output to stdout, text, and JSON, accumulate statistics, and utilize various portable function calls for common needs w.r.t. timers, resource usage, etc.
A tutorial on Kokkos Tools can be found here: https://github.com/kokkos/kokkos-tutorials/blob/main/LectureSeries/KokkosTutorial_07_Tools.pdf
- Vivek Kale (vlkale@sandia.gov)
- Christian Trott (crtrott@sandia.gov)
Special thanks to David Poliakoff on earlier work on Kokkos Tools development.