Definition stage

Check the CUDA API calls made when defining a Kokkos::Experimental::Graph.

class examples.kokkos.graph.example_definition.TestDefinitionView on GitHub

Bases: CMakeAwareTestCase

Trace the CUDA API calls during Kokkos::Experimental::Graph definition stage.

It uses examples/kokkos/graph/example_definition.cpp.

KOKKOS_TOOLS_NVTX_CONNECTOR_LIB

Used in TestNSYS.report().

classmethod get_target_name() → strView on GitHub

class examples.kokkos.graph.example_definition.TestNSYSView on GitHub

Bases: TestDefinition

nsys-focused analysis.

pytestmark = [Mark(name='skipif', args=(True,), kwargs={'reason': 'needs a GPU'})]

report() → ReportView on GitHub

Analyse with nsys, use reprospect.tools.nsys.Cacher.

test_api_calls(report: Report) → NoneView on GitHub

Check the sequence of CUDA API calls to create the graph, nodes and edges.