Kokkos_Profiling_Interface.hpp
No OneTemporary
Actions

Subscribers

None

File Metadata

Created: Thu, Jul 17, 13:10

Kokkos_Profiling_Interface.hpp
View Options

	/*
	//@HEADER
	// ************************************************************************
	//
	// Kokkos v. 2.0
	// Copyright (2014) Sandia Corporation
	//
	// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
	// the U.S. Government retains certain rights in this software.
	//
	// Redistribution and use in source and binary forms, with or without
	// modification, are permitted provided that the following conditions are
	// met:
	//
	// 1. Redistributions of source code must retain the above copyright
	// notice, this list of conditions and the following disclaimer.
	//
	// 2. Redistributions in binary form must reproduce the above copyright
	// notice, this list of conditions and the following disclaimer in the
	// documentation and/or other materials provided with the distribution.
	//
	// 3. Neither the name of the Corporation nor the names of the
	// contributors may be used to endorse or promote products derived from
	// this software without specific prior written permission.
	//
	// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
	// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
	// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
	// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
	// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
	// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
	// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
	// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
	// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
	// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
	// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
	//
	// Questions? Contact H. Carter Edwards (hcedwar@sandia.gov)
	//
	// ************************************************************************
	//@HEADER
	*/

	#ifndef KOKKOSP_INTERFACE_HPP
	#define KOKKOSP_INTERFACE_HPP

	#include <Kokkos_Macros.hpp>
	#if defined(KOKKOS_ENABLE_PROFILING)

	#include <cstddef>
	#include <Kokkos_Core_fwd.hpp>
	#include <string>
	#include <cinttypes>

	#include <impl/Kokkos_Profiling_DeviceInfo.hpp>
	#include <dlfcn.h>
	#include <iostream>
	#include <cstdlib>

	#define KOKKOSP_INTERFACE_VERSION 20150628

	namespace Kokkos {
	namespace Profiling {

	struct SpaceHandle {
	SpaceHandle(const char* space_name);
	char name[64];
	};

	typedef void (*initFunction)(const int,
	const uint64_t,
	const uint32_t,
	KokkosPDeviceInfo*);
	typedef void (*finalizeFunction)();
	typedef void (beginFunction)(const char, const uint32_t, uint64_t*);
	typedef void (*endFunction)(uint64_t);

	typedef void (pushFunction)(const char);
	typedef void (*popFunction)();

	typedef void (allocateDataFunction)(const SpaceHandle, const char, const void*, const uint64_t);
	typedef void (deallocateDataFunction)(const SpaceHandle, const char, const void*, const uint64_t);


	static initFunction initProfileLibrary = NULL;
	static finalizeFunction finalizeProfileLibrary = NULL;

	static beginFunction beginForCallee = NULL;
	static beginFunction beginScanCallee = NULL;
	static beginFunction beginReduceCallee = NULL;
	static endFunction endForCallee = NULL;
	static endFunction endScanCallee = NULL;
	static endFunction endReduceCallee = NULL;

	static pushFunction pushRegionCallee = NULL;
	static popFunction popRegionCallee = NULL;

	static allocateDataFunction allocateDataCallee = NULL;
	static deallocateDataFunction deallocateDataCallee = NULL;


	bool profileLibraryLoaded();

	void beginParallelFor(const std::string& kernelPrefix, const uint32_t devID, uint64_t* kernelID);
	void endParallelFor(const uint64_t kernelID);
	void beginParallelScan(const std::string& kernelPrefix, const uint32_t devID, uint64_t* kernelID);
	void endParallelScan(const uint64_t kernelID);
	void beginParallelReduce(const std::string& kernelPrefix, const uint32_t devID, uint64_t* kernelID);
	void endParallelReduce(const uint64_t kernelID);

	void pushRegion(const std::string& kName);
	void popRegion();

	void allocateData(const SpaceHandle space, const std::string label, const void* ptr, const uint64_t size);
	void deallocateData(const SpaceHandle space, const std::string label, const void* ptr, const uint64_t size);

	void initialize();
	void finalize();

	//Define finalize_fake inline to get rid of warnings for unused static variables
	inline void finalize_fake() {
	if(NULL != finalizeProfileLibrary) {
	(*finalizeProfileLibrary)();

	// Set all profile hooks to NULL to prevent
	// any additional calls. Once we are told to
	// finalize, we mean it
	beginForCallee = NULL;
	beginScanCallee = NULL;
	beginReduceCallee = NULL;
	endScanCallee = NULL;
	endForCallee = NULL;
	endReduceCallee = NULL;

	allocateDataCallee = NULL;
	deallocateDataCallee = NULL;

	initProfileLibrary = NULL;
	finalizeProfileLibrary = NULL;
	pushRegionCallee = NULL;
	popRegionCallee = NULL;
	}
	}


	}
	}

	#endif
	#endif

Kokkos_Profiling_Interface.hppNo OneTemporaryActions

File Metadata

Kokkos_Profiling_Interface.hppView Options

Event Timeline

Kokkos_Profiling_Interface.hpp
No OneTemporary
Actions

Kokkos_Profiling_Interface.hpp
View Options