kokkosp.hpp File Reference

#include "timemory/api.hpp"
#include "timemory/compat/macros.h"
#include "timemory/components/data_tracker/components.hpp"
#include "timemory/components/user_bundle/types.hpp"
#include "timemory/variadic/component_bundle.hpp"
#include <cstdint>
#include <functional>
#include <map>
#include <mutex>
#include <string>
#include <tuple>
#include <unordered_map>
#include <vector>

Include dependency graph for kokkosp.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	tim::kokkosp::kernel_logger

Namespaces
namespace	tim
namespace	tim::kokkosp

Macros
#define	TIMEMORY_KOKKOSP_PREFIX
#define	TIMEMORY_KOKKOSP_POSTFIX   TIMEMORY_VISIBILITY("default")

Typedefs
using	tim::kokkosp::memory_tracker = tim::component::data_tracker< int64_t, tim::project::kokkosp >
using	tim::kokkosp::kokkos_bundle = tim::component::user_kokkosp_bundle
using	tim::kokkosp::logger_t = tim::component_bundle_t< project::kokkosp, kokkosp::kernel_logger * >
template<typename... Tail>
using	tim::kokkosp::profiler_t = tim::component_bundle_t< project::kokkosp, kokkosp::memory_tracker, Tail... >
template<typename... Tail>
using	tim::kokkosp::profiler_section_t = std::tuple< std::string, profiler_t< Tail... > >
template<typename... Tail>
using	tim::kokkosp::profiler_alloc_t = tim::auto_tuple< kokkosp::memory_tracker, Tail... >
template<typename... Tail>
using	tim::kokkosp::profiler_stack_t = std::vector< profiler_t< Tail... > >
template<typename... Tail>
using	tim::kokkosp::profiler_memory_map_t = std::unordered_map< string_view_t, std::unordered_map< string_view_t, profiler_t< Tail... > > >
template<typename... Tail>
using	tim::kokkosp::profiler_index_map_t = std::unordered_map< uint64_t, profiler_t< Tail... > >
template<typename... Tail>
using	tim::kokkosp::profiler_section_map_t = std::unordered_map< uint64_t, profiler_section_t< Tail... > >

Enumerations
enum	tim::kokkosp::Space {   tim::kokkosp::SPACE_HOST ,   tim::kokkosp::SPACE_CUDA }
enum	{ tim::kokkosp::NSPACES = 2 }

Functions
Space	tim::kokkosp::get_space (const SpaceHandle &handle)
const char *	tim::kokkosp::get_space_name (int space)
uint64_t	tim::kokkosp::get_unique_id ()
std::mutex &	tim::kokkosp::get_cleanup_mutex ()
auto &	tim::kokkosp::get_cleanup ()
template<typename Tp >
Tp &	tim::kokkosp::get_tl_static ()
template<typename Tp >
Tp &	tim::kokkosp::get_static ()
void	tim::kokkosp::cleanup ()
template<typename... Tail>
profiler_index_map_t< Tail... > &	tim::kokkosp::get_profiler_index_map ()
template<typename... Tail>
profiler_section_map_t< Tail... > &	tim::kokkosp::get_profiler_section_map ()
template<typename... Tail>
profiler_memory_map_t< Tail... > &	tim::kokkosp::get_profiler_memory_map ()
template<typename... Tail>
auto &	tim::kokkosp::get_profiler_memory_map (SpaceHandle _space)
template<typename... Tail>
profiler_stack_t< Tail... > &	tim::kokkosp::get_profiler_stack ()
template<typename... Tail>
void	tim::kokkosp::create_profiler (const std::string &pname, uint64_t kernid)
template<typename... Tail>
void	tim::kokkosp::destroy_profiler (uint64_t kernid)
template<typename... Tail>
void	tim::kokkosp::start_profiler (uint64_t kernid)
template<typename... Tail>
void	tim::kokkosp::stop_profiler (uint64_t kernid)
	TIMEMORY_DEFINE_CONCRETE_TRAIT (uses_memory_units, kokkosp::memory_tracker, std::true_type) TIMEMORY_DEFINE_CONCRETE_TRAIT(is_memory_category
std::true_type void	kokkosp_print_help (char *argv0)
void	kokkosp_parse_args (int argc, char **argv)
void	kokkosp_declare_metadata (const char *key, const char *value)
void	kokkosp_init_library (const int loadSeq, const uint64_t interfaceVer, const uint32_t devInfoCount, void *deviceInfo)
void	kokkosp_finalize_library ()
void	kokkosp_begin_parallel_for (const char *name, uint32_t devid, uint64_t *kernid)
void	kokkosp_end_parallel_for (uint64_t kernid)
void	kokkosp_begin_parallel_reduce (const char *name, uint32_t devid, uint64_t *kernid)
void	kokkosp_end_parallel_reduce (uint64_t kernid)
void	kokkosp_begin_parallel_scan (const char *name, uint32_t devid, uint64_t *kernid)
void	kokkosp_end_parallel_scan (uint64_t kernid)
void	kokkosp_begin_fence (const char *name, uint32_t devid, uint64_t *kernid)
void	kokkosp_end_fence (uint64_t kernid)
void	kokkosp_push_profile_region (const char *name)
void	kokkosp_pop_profile_region ()
void	kokkosp_create_profile_section (const char *name, uint32_t *secid)
void	kokkosp_destroy_profile_section (uint32_t secid)
void	kokkosp_start_profile_section (uint32_t secid)
void	kokkosp_stop_profile_section (uint32_t secid)
void	kokkosp_allocate_data (const SpaceHandle space, const char *label, const void *const ptr, const uint64_t size)
void	kokkosp_deallocate_data (const SpaceHandle space, const char *label, const void *const ptr, const uint64_t size)
void	kokkosp_begin_deep_copy (SpaceHandle dst_handle, const char *dst_name, const void *dst_ptr, SpaceHandle src_handle, const char *src_name, const void *src_ptr, uint64_t size)
void	kokkosp_end_deep_copy ()
void	kokkosp_profile_event (const char *name)

Macro Definition Documentation

TIMEMORY_KOKKOSP_POSTFIX

#define TIMEMORY_KOKKOSP_POSTFIX   TIMEMORY_VISIBILITY("default")

Definition at line 55 of file kokkosp.hpp.

TIMEMORY_KOKKOSP_PREFIX

#define TIMEMORY_KOKKOSP_PREFIX

Definition at line 46 of file kokkosp.hpp.

Function Documentation

kokkosp_allocate_data()

void kokkosp_allocate_data	(	const SpaceHandle	space,
		const char *	label,
		const void *const	ptr,
		const uint64_t	size
	)

Definition at line 309 of file kokkosp.cpp.

    {
        kokkosp::logger_t{}.mark(0, __FUNCTION__, space.name, label,
                                 TIMEMORY_JOIN("", '[', ptr, ']'), size);
        kokkosp::profiler_alloc_t<>{ TIMEMORY_JOIN('/', "kokkos/allocate", space.name,
                                                   label) }
            .store(std::plus<int64_t>{}, size);
    }

References TIMEMORY_JOIN.

kokkosp_begin_deep_copy()

void kokkosp_begin_deep_copy	(	SpaceHandle	dst_handle,
		const char *	dst_name,
		const void *	dst_ptr,
		SpaceHandle	src_handle,
		const char *	src_name,
		const void *	src_ptr,
		uint64_t	size
	)

Definition at line 331 of file kokkosp.cpp.

    {
        kokkosp::logger_t{}.mark(1, __FUNCTION__, dst_handle.name, dst_name,
                                 TIMEMORY_JOIN("", '[', dst_ptr, ']'), src_handle.name,
                                 src_name, TIMEMORY_JOIN("", '[', src_ptr, ']'), size);
 
        auto name = TIMEMORY_JOIN('/', "kokkos/deep_copy",
                                  TIMEMORY_JOIN('=', dst_handle.name, dst_name),
                                  TIMEMORY_JOIN('=', src_handle.name, src_name));
 
        auto& _data = kokkosp::get_profiler_stack<kokkosp::kokkos_bundle>();
        _data.emplace_back(name);
        _data.back().audit(dst_handle, dst_name, dst_ptr, src_handle, src_name, src_ptr,
                           size);
        _data.back().start();
        _data.back().store(std::plus<int64_t>{}, size);
    }

References TIMEMORY_JOIN.

kokkosp_begin_fence()

void kokkosp_begin_fence	(	const char *	name,
		uint32_t	devid,
		uint64_t *	kernid
	)

Definition at line 241 of file kokkosp.cpp.

    {
        auto pname =
            (devid > std::numeric_limits<uint16_t>::max())  // junk device number
                ? TIMEMORY_JOIN('/', "kokkos", name)
                : TIMEMORY_JOIN('/', "kokkos", TIMEMORY_JOIN("", "dev", devid), name);
        *kernid = kokkosp::get_unique_id();
        kokkosp::logger_t{}.mark(1, __FUNCTION__, name, *kernid);
        kokkosp::create_profiler<kokkosp::kokkos_bundle>(pname, *kernid);
        kokkosp::start_profiler<kokkosp::kokkos_bundle>(*kernid);
    }

References tim::kokkosp::get_unique_id(), std::max(), and TIMEMORY_JOIN.

kokkosp_begin_parallel_for()

void kokkosp_begin_parallel_for	(	const char *	name,
		uint32_t	devid,
		uint64_t *	kernid
	)

Definition at line 178 of file kokkosp.cpp.

    {
        auto pname =
            (devid > std::numeric_limits<uint16_t>::max())  // junk device number
                ? TIMEMORY_JOIN('/', "kokkos", name)
                : TIMEMORY_JOIN('/', "kokkos", TIMEMORY_JOIN("", "dev", devid), name);
        *kernid = kokkosp::get_unique_id();
        kokkosp::logger_t{}.mark(1, __FUNCTION__, name, *kernid);
        kokkosp::create_profiler<kokkosp::kokkos_bundle>(pname, *kernid);
        kokkosp::start_profiler<kokkosp::kokkos_bundle>(*kernid);
    }

References tim::kokkosp::get_unique_id(), std::max(), and TIMEMORY_JOIN.

kokkosp_begin_parallel_reduce()

void kokkosp_begin_parallel_reduce	(	const char *	name,
		uint32_t	devid,
		uint64_t *	kernid
	)

Definition at line 199 of file kokkosp.cpp.

    {
        auto pname =
            (devid > std::numeric_limits<uint16_t>::max())  // junk device number
                ? TIMEMORY_JOIN('/', "kokkos", name)
                : TIMEMORY_JOIN('/', "kokkos", TIMEMORY_JOIN("", "dev", devid), name);
        *kernid = kokkosp::get_unique_id();
        kokkosp::logger_t{}.mark(1, __FUNCTION__, name, *kernid);
        kokkosp::create_profiler<kokkosp::kokkos_bundle>(pname, *kernid);
        kokkosp::start_profiler<kokkosp::kokkos_bundle>(*kernid);
    }

References tim::kokkosp::get_unique_id(), std::max(), and TIMEMORY_JOIN.

kokkosp_begin_parallel_scan()

void kokkosp_begin_parallel_scan	(	const char *	name,
		uint32_t	devid,
		uint64_t *	kernid
	)

Definition at line 220 of file kokkosp.cpp.

    {
        auto pname =
            (devid > std::numeric_limits<uint16_t>::max())  // junk device number
                ? TIMEMORY_JOIN('/', "kokkos", name)
                : TIMEMORY_JOIN('/', "kokkos", TIMEMORY_JOIN("", "dev", devid), name);
        *kernid = kokkosp::get_unique_id();
        kokkosp::logger_t{}.mark(1, __FUNCTION__, name, *kernid);
        kokkosp::create_profiler<kokkosp::kokkos_bundle>(pname, *kernid);
        kokkosp::start_profiler<kokkosp::kokkos_bundle>(*kernid);
    }

References tim::kokkosp::get_unique_id(), std::max(), and TIMEMORY_JOIN.

kokkosp_create_profile_section()

void kokkosp_create_profile_section	(	const char *	name,
		uint32_t *	secid
	)

Definition at line 281 of file kokkosp.cpp.

    {
        *secid     = kokkosp::get_unique_id();
        auto pname = TIMEMORY_JOIN('/', "kokkos", name);
        kokkosp::create_profiler<kokkosp::kokkos_bundle>(pname, *secid);
    }

References tim::kokkosp::get_unique_id(), and TIMEMORY_JOIN.

kokkosp_deallocate_data()

void kokkosp_deallocate_data	(	const SpaceHandle	space,
		const char *	label,
		const void *const	ptr,
		const uint64_t	size
	)

Definition at line 319 of file kokkosp.cpp.

    {
        kokkosp::logger_t{}.mark(0, __FUNCTION__, space.name, label,
                                 TIMEMORY_JOIN("", '[', ptr, ']'), size);
        kokkosp::profiler_alloc_t<>{ TIMEMORY_JOIN('/', "kokkos/deallocate", space.name,
                                                   label) }
            .store(std::plus<int64_t>{}, size);
    }

References TIMEMORY_JOIN.

kokkosp_declare_metadata()

void kokkosp_declare_metadata	(	const char *	key,
		const char *	value
	)

Definition at line 123 of file kokkosp.cpp.

    {
        tim::manager::add_metadata(key, value);
    }

References tim::manager::add_metadata().

kokkosp_destroy_profile_section()

void kokkosp_destroy_profile_section

(

uint32_t

secid

)

Definition at line 288 of file kokkosp.cpp.

    {
        kokkosp::destroy_profiler<kokkosp::kokkos_bundle>(secid);
    }

kokkosp_end_deep_copy()

void kokkosp_end_deep_copy

(

)

Definition at line 351 of file kokkosp.cpp.

    {
        kokkosp::logger_t{}.mark(-1, __FUNCTION__);
        auto& _data = kokkosp::get_profiler_stack<kokkosp::kokkos_bundle>();
        if(_data.empty())
            return;
        _data.back().store(std::minus<int64_t>{}, 0);
        _data.back().stop();
        _data.pop_back();
    }

kokkosp_end_fence()

void kokkosp_end_fence

(

uint64_t

kernid

)

Definition at line 253 of file kokkosp.cpp.

    {
        kokkosp::logger_t{}.mark(-1, __FUNCTION__, kernid);
        kokkosp::stop_profiler<kokkosp::kokkos_bundle>(kernid);
        kokkosp::destroy_profiler<kokkosp::kokkos_bundle>(kernid);
    }

kokkosp_end_parallel_for()

void kokkosp_end_parallel_for

(

uint64_t

kernid

)

Definition at line 190 of file kokkosp.cpp.

    {
        kokkosp::logger_t{}.mark(-1, __FUNCTION__, kernid);
        kokkosp::stop_profiler<kokkosp::kokkos_bundle>(kernid);
        kokkosp::destroy_profiler<kokkosp::kokkos_bundle>(kernid);
    }

kokkosp_end_parallel_reduce()

void kokkosp_end_parallel_reduce

(

uint64_t

kernid

)

Definition at line 211 of file kokkosp.cpp.

    {
        kokkosp::logger_t{}.mark(-1, __FUNCTION__, kernid);
        kokkosp::stop_profiler<kokkosp::kokkos_bundle>(kernid);
        kokkosp::destroy_profiler<kokkosp::kokkos_bundle>(kernid);
    }

kokkosp_end_parallel_scan()

void kokkosp_end_parallel_scan

(

uint64_t

kernid

)

Definition at line 232 of file kokkosp.cpp.

    {
        kokkosp::logger_t{}.mark(-1, __FUNCTION__, kernid);
        kokkosp::stop_profiler<kokkosp::kokkos_bundle>(kernid);
        kokkosp::destroy_profiler<kokkosp::kokkos_bundle>(kernid);
    }

kokkosp_finalize_library()

void kokkosp_finalize_library

(

)

Definition at line 165 of file kokkosp.cpp.

    {
        printf("\n%s\n", kokkos_banner.c_str());
        printf("KokkosP: Finalization of timemory Connector. Complete.\n");
        printf("%s\n\n", kokkos_banner.c_str());
 
        kokkosp::cleanup();
 
        tim::timemory_finalize();
    }

References tim::kokkosp::cleanup(), and tim::timemory_finalize().

kokkosp_init_library()

void kokkosp_init_library	(	const int	loadSeq,
		const uint64_t	interfaceVer,
		const uint32_t	devInfoCount,
		void *	deviceInfo
	)

Definition at line 128 of file kokkosp.cpp.

    {
        add_kernel_logger();
 
        tim::consume_parameters(devInfoCount, deviceInfo);
 
        tim::set_env("TIMEMORY_TIME_OUTPUT", "ON", 0);
        tim::set_env("TIMEMORY_COUT_OUTPUT", "OFF", 0);
        tim::set_env("TIMEMORY_ADD_SECONDARY", "OFF", 0);
 
        printf("%s\n", kokkos_banner.c_str());
        printf("# KokkosP: timemory Connector (sequence is %d, version: %llu)\n", loadSeq,
               (unsigned long long) interfaceVer);
        printf("%s\n\n", kokkos_banner.c_str());
 
        if(tim::settings::output_path() == "timemory-output")
        {
            // timemory_init is expecting some args so generate some
            std::array<char*, 1> cstr = { { strdup("kokkosp") } };
            tim::timemory_init(1, cstr.data());
            free(cstr[0]);
        }
        assert(env_configured);
 
        // search unique and fallback environment variables
        namespace operation = tim::operation;
        operation::init<kokkosp::kokkos_bundle>(
            operation::mode_constant<operation::init_mode::global>{});
 
        // add at least one
        if(kokkosp::kokkos_bundle::bundle_size() == 0)
            kokkosp::kokkos_bundle::configure<tim::component::wall_clock>();
 
        setup_kernel_logger();
    }

References tim::consume_parameters(), tim::output_path, tim::set_env(), strdup(), and tim::timemory_init().

kokkosp_parse_args()

void kokkosp_parse_args	(	int	argc,
		char **	argv
	)

Definition at line 107 of file kokkosp.cpp.

    {
        add_kernel_logger();
 
        tim::timemory_init(argc, argv);
 
        std::vector<std::string> _args{};
        _args.reserve(argc);
        for(int i = 0; i < argc; ++i)
            _args.emplace_back(argv[i]);
 
        tim::timemory_argparse(_args);
 
        setup_kernel_logger();
    }

References tim::_args, tim::argv, tim::timemory_argparse(), and tim::timemory_init().

kokkosp_pop_profile_region()

void kokkosp_pop_profile_region

(

)

Definition at line 270 of file kokkosp.cpp.

    {
        kokkosp::logger_t{}.mark(-1, __FUNCTION__);
        if(kokkosp::get_profiler_stack<kokkosp::kokkos_bundle>().empty())
            return;
        kokkosp::get_profiler_stack<kokkosp::kokkos_bundle>().back().stop();
        kokkosp::get_profiler_stack<kokkosp::kokkos_bundle>().pop_back();
    }

kokkosp_print_help()

std::true_type void kokkosp_print_help

(

char *

argv0

)

Definition at line 100 of file kokkosp.cpp.

    {
        add_kernel_logger();
        std::vector<std::string> _args = { argv, "--help" };
        tim::timemory_argparse(_args);
    }

References tim::_args, tim::argv, and tim::timemory_argparse().

kokkosp_profile_event()

void kokkosp_profile_event

(

const char *

name

)

Definition at line 364 of file kokkosp.cpp.

    {
        kokkosp::profiler_t<kokkosp::kokkos_bundle>{}.mark(name);
    }

kokkosp_push_profile_region()

void kokkosp_push_profile_region

(

const char *

name

)

Definition at line 262 of file kokkosp.cpp.

    {
        kokkosp::logger_t{}.mark(1, __FUNCTION__, name);
        kokkosp::get_profiler_stack<kokkosp::kokkos_bundle>().push_back(
            kokkosp::profiler_t<kokkosp::kokkos_bundle>(name));
        kokkosp::get_profiler_stack<kokkosp::kokkos_bundle>().back().start();
    }

kokkosp_start_profile_section()

void kokkosp_start_profile_section

(

uint32_t

secid

)

Definition at line 295 of file kokkosp.cpp.

    {
        kokkosp::logger_t{}.mark(1, __FUNCTION__, secid);
        kokkosp::start_profiler<kokkosp::kokkos_bundle>(secid);
    }

kokkosp_stop_profile_section()

void kokkosp_stop_profile_section

(

uint32_t

secid

)

Definition at line 301 of file kokkosp.cpp.

    {
        kokkosp::logger_t{}.mark(-1, __FUNCTION__, secid);
        kokkosp::start_profiler<kokkosp::kokkos_bundle>(secid);
    }

TIMEMORY_DEFINE_CONCRETE_TRAIT()

TIMEMORY_DEFINE_CONCRETE_TRAIT	(	uses_memory_units	,
		kokkosp::memory_tracker	,
		std::true_type
	)