foreman
119 строки
3.1 KiB
C++
119 строки
3.1 KiB
C++
//
|
|
// Copyright (c) 2011 Advanced Micro Devices, Inc. All rights reserved.
|
|
//
|
|
|
|
#ifndef CPUPROGRAM_HPP_
|
|
#define CPUPROGRAM_HPP_
|
|
|
|
#include "top.hpp"
|
|
#include "device/device.hpp"
|
|
#include "device/cpu/cpubinary.hpp"
|
|
#include <string>
|
|
|
|
#include "jit.h"
|
|
|
|
// forward declaration
|
|
namespace amd {
|
|
namespace option {
|
|
class Options;
|
|
} // option
|
|
} // amd
|
|
|
|
//! \namespace cpu CPU Device Implementation
|
|
namespace cpu {
|
|
|
|
//! \class CPU program
|
|
class Program : public device::Program
|
|
{
|
|
private:
|
|
aclJITObjectImage JITBinary;
|
|
std::string sourceFileName_; //!< The source image.
|
|
void* handle_; // @todo: remove me
|
|
|
|
public:
|
|
//! Default constructor
|
|
Program(Device& cpuDev)
|
|
: device::Program(cpuDev), JITBinary(NULL), handle_(NULL) {}
|
|
|
|
//! Default destructor
|
|
~Program();
|
|
|
|
//! pre-compile setup for CPU
|
|
virtual bool initBuild(amd::option::Options* options);
|
|
|
|
//! post-compile setup for CPU
|
|
virtual bool finiBuild(bool isBuildGood);
|
|
|
|
//! Compiles CPU program
|
|
virtual bool compileImpl(
|
|
const std::string& sourceCode,
|
|
const std::vector<const std::string*>& headers,
|
|
const char** headerIncludeNames,
|
|
amd::option::Options* options );
|
|
|
|
//! Links CPU program
|
|
virtual bool linkImpl(amd::option::Options* options = NULL);
|
|
|
|
//! Links CPU programs
|
|
virtual bool linkImpl(
|
|
const std::vector<device::Program*>& inputPrograms,
|
|
amd::option::Options* options = NULL,
|
|
bool createLibrary = false);
|
|
|
|
virtual bool createBinary(amd::option::Options* options);
|
|
|
|
//! Returns the device object, associated with this program.
|
|
const Device& device() {
|
|
return static_cast<const Device&>(device::Program::device());
|
|
}
|
|
|
|
/*! \brief Invokes the LLC compiler for the LLVM binary compilation
|
|
* to x86 ASM text source code and ISA binary
|
|
*
|
|
* \return True if we successefully compiled a CPU program
|
|
*/
|
|
bool compileBinaryToISA(
|
|
amd::option::Options* options //!< options for compilation
|
|
);
|
|
|
|
//! Load the library into memory
|
|
bool loadDllCode(amd::option::Options* options, bool addElfSymbols=false);
|
|
|
|
//! Initialize binary for CPU
|
|
virtual bool initClBinary();
|
|
|
|
//! Release binary for CPU
|
|
virtual void releaseClBinary();
|
|
|
|
ClBinary* clBinary() {
|
|
return static_cast<ClBinary*>(device::Program::clBinary());
|
|
}
|
|
const ClBinary* clBinary() const {
|
|
return static_cast<const ClBinary*>(device::Program::clBinary());
|
|
}
|
|
|
|
aclJITObjectImage getJITBinary() { return this->JITBinary; }
|
|
void setJITBinary(aclJITObjectImage JITBinary) { this->JITBinary = JITBinary; }
|
|
|
|
private:
|
|
aclCompiler* compiler() { return static_cast<const Device&>(device()).compiler(); }
|
|
|
|
//! Disable default copy constructor
|
|
Program(const Program&);
|
|
|
|
//! Disable operator=
|
|
Program& operator=(const Program&);
|
|
|
|
std::string dllFileName_; //!< File name of the dll with kernels
|
|
protected:
|
|
virtual bool isElf(const char* bin) const {
|
|
return amd::isElfHeader(bin, LP64_SWITCH(ELFCLASS32, ELFCLASS64));
|
|
}
|
|
|
|
virtual const aclTargetInfo & info(const char * str = "");
|
|
};
|
|
|
|
} // namespace cpu
|
|
|
|
#endif // CPUPROGRAM_HPP_
|