mirror of
https://github.com/ldc-developers/ldc.git
synced 2025-05-03 16:41:06 +03:00
81 lines
2.4 KiB
C++
81 lines
2.4 KiB
C++
//===-- driver/dcomputecodegenerator.cpp ----------------------------------===//
|
||
//
|
||
// LDC – the LLVM D compiler
|
||
//
|
||
// This file is distributed under the BSD-style LDC license. See the LICENSE
|
||
// file for details.
|
||
//
|
||
//===----------------------------------------------------------------------===//
|
||
|
||
#include "driver/dcomputecodegenerator.h"
|
||
#include "driver/cl_options.h"
|
||
#include "ddmd/errors.h"
|
||
#include "gen/cl_helpers.h"
|
||
#include "ir/irdsymbol.h"
|
||
#include "llvm/Support/CommandLine.h"
|
||
#include <array>
|
||
#include <string>
|
||
#include <algorithm>
|
||
|
||
DComputeTarget *
|
||
DComputeCodeGenManager::createComputeTarget(const std::string &s) {
|
||
int v;
|
||
#define OCL_VALID_VER_INIT 100, 110, 120, 200, 210, 220
|
||
const std::array<int, 6> valid_ocl_versions = {{OCL_VALID_VER_INIT}};
|
||
#define CUDA_VALID_VER_INIT 100, 110, 120, 130, 200, 210, 300, 350, 370,\
|
||
500, 520, 600, 610, 620
|
||
const std::array<int, 14> vaild_cuda_versions = {{CUDA_VALID_VER_INIT}};
|
||
|
||
if (s.substr(0, 4) == "ocl-") {
|
||
v = atoi(s.c_str() + 4);
|
||
if (std::find(valid_ocl_versions.begin(), valid_ocl_versions.end(), v) !=
|
||
valid_ocl_versions.end()) {
|
||
return createOCLTarget(ctx, v);
|
||
}
|
||
} else if (s.substr(0, 5) == "cuda-") {
|
||
v = atoi(s.c_str() + 5);
|
||
|
||
if (std::find(vaild_cuda_versions.begin(), vaild_cuda_versions.end(), v) !=
|
||
vaild_cuda_versions.end()) {
|
||
return createCUDATarget(ctx, v);
|
||
}
|
||
}
|
||
#define XSTR(x) #x
|
||
#define STR(x) XSTR((x))
|
||
|
||
error(Loc(),
|
||
"unrecognised or invalid DCompute targets: the format is ocl-xy0 "
|
||
"for OpenCl x.y and cuda-xy0 for CUDA CC x.y. Valid versions "
|
||
"for OpenCl are " STR(OCL_VALID_VER_INIT) ". Valid versions for CUDA "
|
||
"are " STR(CUDA_VALID_VER_INIT));
|
||
fatal();
|
||
return nullptr;
|
||
}
|
||
|
||
DComputeCodeGenManager::DComputeCodeGenManager(llvm::LLVMContext &c) : ctx(c) {
|
||
#if LDC_LLVM_SUPPORTED_TARGET_SPIRV || LDC_LLVM_SUPPORTED_TARGET_NVPTX
|
||
for (auto &option : opts::dcomputeTargets) {
|
||
targets.push_back(createComputeTarget(option));
|
||
}
|
||
#endif
|
||
oldGIR = gIR;
|
||
oldGTargetMachine = gTargetMachine;
|
||
}
|
||
|
||
void DComputeCodeGenManager::emit(Module *m) {
|
||
for (auto &target : targets) {
|
||
target->emit(m);
|
||
IrDsymbol::resetAll();
|
||
}
|
||
}
|
||
|
||
void DComputeCodeGenManager::writeModules() {
|
||
for (auto &target : targets) {
|
||
target->writeModule();
|
||
}
|
||
}
|
||
|
||
DComputeCodeGenManager::~DComputeCodeGenManager() {
|
||
gIR = oldGIR;
|
||
gTargetMachine = oldGTargetMachine;
|
||
}
|