ldc/gen/dcompute/target.h
2022-09-15 20:00:52 +08:00

61 lines
1.7 KiB
C++
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

//===-- gen/dcompute/target.h ------------------------------------*- C++-*-===//
//
// LDC the LLVM D compiler
//
// This file is distributed under the BSD-style LDC license. See the LICENSE
// file for details.
//
//===----------------------------------------------------------------------===//
#pragma once
#include "gen/irstate.h"
#include "llvm/IR/LLVMContext.h"
#include "llvm/IR/Function.h"
#include <array>
namespace llvm {
class Module;
class Function;
class TargetMachine;
}
class Module;
class FuncDeclaration;
class DComputeTarget {
public:
llvm::LLVMContext &ctx;
int tversion; // OpenCL or CUDA CC version:major*100 + minor*10
enum class ID { Host = 0, OpenCL = 1, CUDA = 2 };
ID target; // ID for codegen time conditional compilation.
const char *short_name;
const char *binSuffix;
TargetABI *abi;
llvm::TargetMachine *targetMachine = nullptr;
// The nominal address spaces in DCompute are Private = 0, Global = 1,
// Shared = 2, Constant = 3, Generic = 4
std::array<int, 5> mapping;
IRState *_ir;
DComputeTarget(llvm::LLVMContext &c, int v, ID id, const char *_short_name,
const char *suffix, TargetABI *a, std::array<int, 5> map)
: ctx(c), tversion(v), target(id), short_name(_short_name),
binSuffix(suffix), abi(a), mapping(map), _ir(nullptr) {}
void emit(Module *m);
void doCodeGen(Module *m);
void writeModule();
virtual void addMetadata() = 0;
virtual void addKernelMetadata(FuncDeclaration *df, llvm::Function *llf) = 0;
};
#if LDC_LLVM_SUPPORTED_TARGET_NVPTX
DComputeTarget *createCUDATarget(llvm::LLVMContext &c, int sm);
#endif
#if LDC_LLVM_SUPPORTED_TARGET_SPIRV
DComputeTarget *createOCLTarget(llvm::LLVMContext &c, int oclver);
#endif