Proteus
Programmable JIT compilation and optimization for C/C++ using LLVM
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Pages
Loading...
Searching...
No Matches
JitEngineDeviceHIP.hpp
Go to the documentation of this file.
1//===-- JitEngineDeviceHIP.hpp -- JIT Engine Device for HIP header --===//
2//
3// Part of the Proteus Project, under the Apache License v2.0 with LLVM
4// Exceptions. See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9//===----------------------------------------------------------------------===//
10
11#ifndef PROTEUS_JITENGINEDEVICEHIP_HPP
12#define PROTEUS_JITENGINEDEVICEHIP_HPP
13
15#include "proteus/Utils.h"
16
17namespace proteus {
18
19using namespace llvm;
20
21class JitEngineDeviceHIP;
22template <> struct DeviceTraits<JitEngineDeviceHIP> {
23 using DeviceError_t = hipError_t;
24 using DeviceStream_t = hipStream_t;
25 using KernelFunction_t = hipFunction_t;
26};
27
28class JitEngineDeviceHIP : public JitEngineDevice<JitEngineDeviceHIP> {
29public:
31
32 static const SmallVector<StringRef> gridDimXFnName() {
33 return {"_ZNK17__HIP_CoordinatesI13__HIP_GridDimE3__XcvjEv",
34 "llvm.amdgcn.num.workgroups.x"};
35 };
36
37 static const SmallVector<StringRef> gridDimYFnName() {
38 return {"_ZNK17__HIP_CoordinatesI13__HIP_GridDimE3__YcvjEv",
39 "llvm.amdgcn.num.workgroups.y"};
40 };
41
42 static const SmallVector<StringRef> gridDimZFnName() {
43 return {"_ZNK17__HIP_CoordinatesI13__HIP_GridDimE3__ZcvjEv",
44 "llvm.amdgcn.num.workgroups.z"};
45 };
46
47 static const SmallVector<StringRef> blockDimXFnName() {
48 return {"_ZNK17__HIP_CoordinatesI14__HIP_BlockDimE3__XcvjEv",
49 "llvm.amdgcn.workgroup.size.x"};
50 };
51
52 static const SmallVector<StringRef> blockDimYFnName() {
53 return {"_ZNK17__HIP_CoordinatesI14__HIP_BlockDimE3__YcvjEv",
54 "llvm.amdgcn.workgroup.size.y"};
55 }
56
57 static const SmallVector<StringRef> blockDimZFnName() {
58 return {"_ZNK17__HIP_CoordinatesI14__HIP_BlockDimE3__ZcvjEv",
59 "llvm.amdgcn.workgroup.size.z"};
60 };
61
62 static const SmallVector<StringRef> blockIdxXFnName() {
63 return {"_ZNK17__HIP_CoordinatesI14__HIP_BlockIdxE3__XcvjEv",
64 "llvm.amdgcn.workgroup.id.x"};
65 };
66
67 static const SmallVector<StringRef> blockIdxYFnName() {
68 return {"_ZNK17__HIP_CoordinatesI14__HIP_BlockIdxE3__YcvjEv",
69 "llvm.amdgcn.workgroup.id.y"};
70 };
71
72 static const SmallVector<StringRef> blockIdxZFnName() {
73 return {"_ZNK17__HIP_CoordinatesI14__HIP_BlockIdxE3__ZcvjEv",
74 "llvm.amdgcn.workgroup.id.z"};
75 }
76
77 static const SmallVector<StringRef> threadIdxXFnName() {
78 return {"_ZNK17__HIP_CoordinatesI15__HIP_ThreadIdxE3__XcvjEv",
79 "llvm.amdgcn.workitem.id.x"};
80 };
81
82 static const SmallVector<StringRef> threadIdxYFnName() {
83 return {"_ZNK17__HIP_CoordinatesI15__HIP_ThreadIdxE3__YcvjEv",
84 "llvm.amdgcn.workitem.id.y"};
85 };
86
87 static const SmallVector<StringRef> threadIdxZFnName() {
88 return {"_ZNK17__HIP_CoordinatesI15__HIP_ThreadIdxE3__ZcvjEv",
89 "llvm.amdgcn.workitem.id.z"};
90 };
91
92 void *resolveDeviceGlobalAddr(const void *Addr);
93
94 void setLaunchBoundsForKernel(Module &M, Function &F, size_t GridSize,
95 int BlockSize);
96
97 void setKernelDims(Module &M, dim3 &GridDim, dim3 &BlockDim);
98
99 std::unique_ptr<Module> extractModule(BinaryInfo &BinInfo);
100
101 std::unique_ptr<MemoryBuffer> codegenObject(Module &M, StringRef DeviceArch);
102
103 hipFunction_t getKernelFunctionFromImage(StringRef KernelName,
104 const void *Image);
105
106 hipError_t launchKernelFunction(hipFunction_t KernelFunc, dim3 GridDim,
107 dim3 BlockDim, void **KernelArgs,
108 uint64_t ShmemSize, hipStream_t Stream);
109
111
112private:
116};
117
118} // namespace proteus
119
120#endif
void char * KernelName
Definition CompilerInterfaceDevice.cpp:50
Definition JitEngineDevice.hpp:78
Definition Hashing.hpp:19
Definition JitEngineDeviceHIP.hpp:28
static const SmallVector< StringRef > blockIdxXFnName()
Definition JitEngineDeviceHIP.hpp:62
void setKernelDims(Module &M, dim3 &GridDim, dim3 &BlockDim)
void setLaunchBoundsForKernel(Module &M, Function &F, size_t GridSize, int BlockSize)
Definition JitEngineDeviceHIP.cpp:208
hipError_t launchKernelFunction(hipFunction_t KernelFunc, dim3 GridDim, dim3 BlockDim, void **KernelArgs, uint64_t ShmemSize, hipStream_t Stream)
Definition JitEngineDeviceHIP.cpp:229
static const SmallVector< StringRef > threadIdxYFnName()
Definition JitEngineDeviceHIP.hpp:82
static const SmallVector< StringRef > threadIdxXFnName()
Definition JitEngineDeviceHIP.hpp:77
hipFunction_t getKernelFunctionFromImage(StringRef KernelName, const void *Image)
Definition JitEngineDeviceHIP.cpp:222
static const SmallVector< StringRef > blockDimXFnName()
Definition JitEngineDeviceHIP.hpp:47
static const SmallVector< StringRef > threadIdxZFnName()
Definition JitEngineDeviceHIP.hpp:87
static const SmallVector< StringRef > blockDimZFnName()
Definition JitEngineDeviceHIP.hpp:57
static const SmallVector< StringRef > blockIdxYFnName()
Definition JitEngineDeviceHIP.hpp:67
std::unique_ptr< Module > extractModule(BinaryInfo &BinInfo)
Definition JitEngineDeviceHIP.cpp:148
void * resolveDeviceGlobalAddr(const void *Addr)
Definition JitEngineDeviceHIP.cpp:29
static const SmallVector< StringRef > gridDimXFnName()
Definition JitEngineDeviceHIP.hpp:32
static const SmallVector< StringRef > blockIdxZFnName()
Definition JitEngineDeviceHIP.hpp:72
static const SmallVector< StringRef > gridDimYFnName()
Definition JitEngineDeviceHIP.hpp:37
std::unique_ptr< MemoryBuffer > codegenObject(Module &M, StringRef DeviceArch)
Definition JitEngineDeviceHIP.cpp:215
static const SmallVector< StringRef > gridDimZFnName()
Definition JitEngineDeviceHIP.hpp:42
static JitEngineDeviceHIP & instance()
Definition JitEngineDeviceHIP.cpp:33
static const SmallVector< StringRef > blockDimYFnName()
Definition JitEngineDeviceHIP.hpp:52
HashT getModuleHash(BinaryInfo &BinInfo)
Definition JitEngineDeviceHIP.cpp:92
Definition JitEngineDevice.hpp:176
std::string DeviceArch
Definition JitEngineDevice.hpp:385
Definition JitEngine.cpp:20
hipFunction_t KernelFunction_t
Definition JitEngineDeviceHIP.hpp:25
hipError_t DeviceError_t
Definition JitEngineDeviceHIP.hpp:23
hipStream_t DeviceStream_t
Definition JitEngineDeviceHIP.hpp:24
Definition JitEngineDevice.hpp:174