foreman
@@ -0,0 +1,200 @@
|
||||
//
|
||||
// Copyright (c) 2008 Advanced Micro Devices, Inc. All rights reserved.
|
||||
//
|
||||
|
||||
/*! \file program.hpp
|
||||
* \brief Declarations for Program and ProgramBinary objects.
|
||||
*
|
||||
* \author Laurent Morichetti (laurent.morichetti@amd.com)
|
||||
* \date October 2008
|
||||
*/
|
||||
|
||||
#ifndef PROGRAM_HPP_
|
||||
#define PROGRAM_HPP_
|
||||
|
||||
#include "top.hpp"
|
||||
#include "device/device.hpp"
|
||||
#include "platform/object.hpp"
|
||||
#include "platform/kernel.hpp"
|
||||
|
||||
#include <set>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include <map>
|
||||
#include <utility>
|
||||
|
||||
namespace amd {
|
||||
|
||||
/*! \addtogroup Runtime
|
||||
* @{
|
||||
*
|
||||
* \addtogroup Program Programs and Kernel functions
|
||||
* @{
|
||||
*/
|
||||
|
||||
//! A kernel function symbol
|
||||
class Symbol : public HeapObject
|
||||
{
|
||||
public:
|
||||
typedef std::map<const Device*, const device::Kernel*> devicekernels_t;
|
||||
|
||||
private:
|
||||
devicekernels_t deviceKernels_; //! All device kernels objects.
|
||||
devicekernels_t devKernelsNoOpt_; //! Kernel objects without optimization
|
||||
KernelSignature signature_; //! Kernel signature.
|
||||
|
||||
public:
|
||||
//! Default constructor
|
||||
Symbol() {}
|
||||
|
||||
//! Set the entry point and check or set the signature.
|
||||
bool setDeviceKernel(
|
||||
const Device& device, //!< Device object.
|
||||
const device::Kernel* func, //!< Device kernel object.
|
||||
bool noAlias = true //!< No-alias optimization
|
||||
);
|
||||
|
||||
//! Return the device kernel.
|
||||
const device::Kernel* getDeviceKernel(
|
||||
const Device& device, //!< Device object.
|
||||
bool noAlias = true //!< No-alias optimization
|
||||
) const;
|
||||
|
||||
//! Return this Symbol's signature.
|
||||
const KernelSignature& signature() const { return signature_; }
|
||||
};
|
||||
|
||||
class Context;
|
||||
|
||||
//! A collection of binaries for devices in the associated context.
|
||||
class Program : public RuntimeObject
|
||||
{
|
||||
public:
|
||||
typedef std::pair<uint8_t*, size_t> binary_t;
|
||||
typedef std::set<Device const*> devicelist_t;
|
||||
typedef std::map<Device const*, binary_t> devicebinary_t;
|
||||
typedef std::map<Device const*, device::Program*> deviceprograms_t;
|
||||
typedef std::map<std::string, Symbol> symbols_t;
|
||||
|
||||
private:
|
||||
//! The context this program is part of.
|
||||
SharedReference<Context> context_;
|
||||
|
||||
std::string sourceCode_; //!< Strings that make up the source code
|
||||
devicebinary_t binary_; //!< The binary image, provided by the app
|
||||
symbols_t* symbolTable_; //!< The program's kernels symbol table
|
||||
std::string kernelNames_; //!< The program kernel names
|
||||
|
||||
//! The device program objects included in this program
|
||||
deviceprograms_t devicePrograms_;
|
||||
deviceprograms_t devProgramsNoOpt_;
|
||||
devicelist_t deviceList_;
|
||||
|
||||
std::string programLog_; //!< Log for parsing options, etc.
|
||||
bool firstBuildNoOpt_; //!< Build program without optimizations
|
||||
|
||||
protected:
|
||||
//! Destroy this program.
|
||||
~Program();
|
||||
|
||||
//! Clears the program object if the app attempts to rebuild the program
|
||||
void clear();
|
||||
|
||||
//! Global build lock (remove when LLVM is thread-safe).
|
||||
static Monitor buildLock_;
|
||||
|
||||
public:
|
||||
//! Construct a new program to be compiled from the given source code.
|
||||
Program(Context& context, const std::string& sourceCode)
|
||||
: context_(context)
|
||||
, sourceCode_(sourceCode)
|
||||
, symbolTable_(NULL)
|
||||
, programLog_()
|
||||
, firstBuildNoOpt_(true)
|
||||
{ }
|
||||
|
||||
//! Construct a new program associated with a context.
|
||||
Program(Context& context)
|
||||
: context_(context)
|
||||
, symbolTable_(NULL)
|
||||
{ }
|
||||
|
||||
//! Returns context, associated with the current program.
|
||||
const Context& context() const { return context_(); }
|
||||
|
||||
//! Return the sections for this program.
|
||||
const deviceprograms_t& devicePrograms() const { return devicePrograms_; }
|
||||
|
||||
//! Return the associated devices.
|
||||
const devicelist_t& deviceList() const { return deviceList_; }
|
||||
|
||||
//! Return the symbols for this program.
|
||||
const symbols_t& symbols() const { return *symbolTable_; }
|
||||
|
||||
//! Return the program source code.
|
||||
const std::string& sourceCode() const { return sourceCode_; }
|
||||
|
||||
//! Return the program log.
|
||||
const std::string& programLog() const { return programLog_; }
|
||||
|
||||
//! Add a binary image to this program.
|
||||
cl_int addDeviceProgram(Device&, const void* image = NULL,
|
||||
size_t len = 0, int oclVer = 120);
|
||||
|
||||
//! Find the section for the given device. Return NULL if not found.
|
||||
device::Program* getDeviceProgram(const Device& device) const;
|
||||
|
||||
//! Return the symbol for the given kernel name.
|
||||
const Symbol* findSymbol(const char* name) const;
|
||||
|
||||
//! Return the binary image.
|
||||
const binary_t& binary(const Device& device) {
|
||||
return binary_[&device.rootDevice()];
|
||||
}
|
||||
|
||||
//! Return the program kernel names
|
||||
const std::string& kernelNames() const { return kernelNames_; }
|
||||
|
||||
//! Compile the program for the given devices.
|
||||
cl_int compile(
|
||||
const std::vector<Device*>& devices,
|
||||
size_t numHeaders,
|
||||
const std::vector<const Program*>& headerPrograms,
|
||||
const char** headerIncludeNames,
|
||||
const char* options = NULL,
|
||||
void (CL_CALLBACK * notifyFptr)(cl_program, void *) = NULL,
|
||||
void* data = NULL,
|
||||
bool optionChangable = true);
|
||||
|
||||
//! Link the programs for the given devices.
|
||||
cl_int link(
|
||||
const std::vector<Device*>& devices,
|
||||
size_t numInputs,
|
||||
const std::vector<Program*>& inputPrograms,
|
||||
const char* options = NULL,
|
||||
void (CL_CALLBACK * notifyFptr)(cl_program, void *) = NULL,
|
||||
void* data = NULL,
|
||||
bool optionChangable = true);
|
||||
|
||||
//! Build the program for the given devices.
|
||||
cl_int build(
|
||||
const std::vector<Device*>& devices,
|
||||
const char* options = NULL,
|
||||
void (CL_CALLBACK * notifyFptr)(cl_program, void *) = NULL,
|
||||
void* data = NULL,
|
||||
bool optionChangable = true);
|
||||
|
||||
//! Build the program for the given devices without noalias optimization
|
||||
bool buildNoOpt(const Device& device, const std::string& kernelName);
|
||||
|
||||
//! RTTI internal implementation
|
||||
virtual ObjectType objectType() const {return ObjectTypeProgram;}
|
||||
};
|
||||
|
||||
/*! @}
|
||||
* @}
|
||||
*/
|
||||
|
||||
} // namespace amd
|
||||
|
||||
#endif /*PROGRAM_HPP_*/
|
||||
Посилання в новій задачі
Заблокувати користувача