133 lines
5.3 KiB
C++
133 lines
5.3 KiB
C++
/*
|
|
* Copyright (c) 2017, Intel Corporation
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a
|
|
* copy of this software and associated documentation files (the "Software"),
|
|
* to deal in the Software without restriction, including without limitation
|
|
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
* and/or sell copies of the Software, and to permit persons to whom the
|
|
* Software is furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included
|
|
* in all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
|
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
|
|
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
|
|
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
|
|
* OTHER DEALINGS IN THE SOFTWARE.
|
|
*/
|
|
|
|
#pragma once
|
|
#include "cif/common/cif_main.h"
|
|
#include "config.h"
|
|
#include "ocl_igc_interface/code_type.h"
|
|
#include "ocl_igc_interface/igc_ocl_device_ctx.h"
|
|
#include "ocl_igc_interface/fcl_ocl_device_ctx.h"
|
|
#include "runtime/built_ins/sip.h"
|
|
#include "runtime/compiler_interface/binary_cache.h"
|
|
#include "runtime/os_interface/os_library.h"
|
|
|
|
#include "CL/cl_platform.h"
|
|
#include <map>
|
|
#include <mutex>
|
|
|
|
namespace OCLRT {
|
|
class Device;
|
|
class Program;
|
|
|
|
struct TranslationArgs {
|
|
char *pInput = nullptr; // data to be translated
|
|
uint32_t InputSize = 0; // size of data to be translated
|
|
const char *pOptions = nullptr; // list of build/compile options
|
|
uint32_t OptionsSize = 0; // size of options list
|
|
const char *pInternalOptions = nullptr; // list of build/compile options
|
|
uint32_t InternalOptionsSize = 0; // size of options list
|
|
void *pTracingOptions = nullptr; // instrumentation options
|
|
uint32_t TracingOptionsCount = 0; // number of instrumentation options
|
|
void *GTPinInput = nullptr; // input structure for GTPin requests
|
|
};
|
|
|
|
class CompilerInterface {
|
|
public:
|
|
CompilerInterface(const CompilerInterface &) = delete;
|
|
CompilerInterface &operator=(const CompilerInterface &) = delete;
|
|
|
|
static CompilerInterface *getInstance() {
|
|
if (pInstance == nullptr) {
|
|
std::lock_guard<std::mutex> guard(mtx);
|
|
if (pInstance == nullptr) {
|
|
auto instance = new CompilerInterface();
|
|
|
|
if (!instance->initialize()) {
|
|
delete instance;
|
|
instance = nullptr;
|
|
}
|
|
pInstance = instance;
|
|
}
|
|
}
|
|
return pInstance;
|
|
}
|
|
|
|
static void shutdown() {
|
|
std::unique_lock<std::mutex> destructionLock(CompilerInterface::mtx);
|
|
if (pInstance) {
|
|
delete pInstance;
|
|
pInstance = nullptr;
|
|
}
|
|
}
|
|
|
|
MOCKABLE_VIRTUAL cl_int build(Program &program, const TranslationArgs &pInputArgs, bool enableCaching);
|
|
|
|
MOCKABLE_VIRTUAL cl_int compile(Program &program, const TranslationArgs &pInputArgs);
|
|
|
|
MOCKABLE_VIRTUAL cl_int link(Program &program, const TranslationArgs &pInputArgs);
|
|
|
|
cl_int createLibrary(Program &program, const TranslationArgs &pInputArgs);
|
|
|
|
MOCKABLE_VIRTUAL cl_int getSipKernelBinary(SipKernelType kernel, const Device &device, std::vector<char> &retBinary);
|
|
|
|
BinaryCache *replaceBinaryCache(BinaryCache *newCache);
|
|
|
|
protected:
|
|
CompilerInterface();
|
|
virtual ~CompilerInterface();
|
|
|
|
bool initialize();
|
|
|
|
static std::mutex mtx;
|
|
MOCKABLE_VIRTUAL std::unique_lock<std::mutex> lock() {
|
|
return std::unique_lock<std::mutex>{mtx};
|
|
}
|
|
std::unique_ptr<BinaryCache> cache = nullptr;
|
|
|
|
static bool useLlvmText;
|
|
static CompilerInterface *pInstance;
|
|
|
|
using igcDevCtxUptr = CIF::RAII::UPtr_t<IGC::IgcOclDeviceCtxTagOCL>;
|
|
using fclDevCtxUptr = CIF::RAII::UPtr_t<IGC::FclOclDeviceCtxTagOCL>;
|
|
|
|
std::unique_ptr<OsLibrary> igcLib;
|
|
CIF::RAII::UPtr_t<CIF::CIFMain> igcMain = nullptr;
|
|
std::map<const Device *, igcDevCtxUptr> igcDeviceContexts;
|
|
|
|
std::unique_ptr<OsLibrary> fclLib;
|
|
CIF::RAII::UPtr_t<CIF::CIFMain> fclMain = nullptr;
|
|
std::map<const Device *, fclDevCtxUptr> fclDeviceContexts;
|
|
CIF::RAII::UPtr_t<IGC::FclOclTranslationCtxTagOCL> fclBaseTranslationCtx = nullptr;
|
|
|
|
MOCKABLE_VIRTUAL CIF::RAII::UPtr_t<IGC::FclOclTranslationCtxTagOCL> createFclTranslationCtx(const Device &device,
|
|
IGC::CodeType::CodeType_t inType,
|
|
IGC::CodeType::CodeType_t outType);
|
|
MOCKABLE_VIRTUAL CIF::RAII::UPtr_t<IGC::IgcOclTranslationCtxTagOCL> createIgcTranslationCtx(const Device &device,
|
|
IGC::CodeType::CodeType_t inType,
|
|
IGC::CodeType::CodeType_t outType);
|
|
|
|
bool isCompilerAvailable() const {
|
|
return (fclMain != nullptr) && (igcMain != nullptr);
|
|
}
|
|
};
|
|
} // namespace OCLRT
|