source: icGREP/icgrep-devel/icgrep/IDISA/idisa_builder.h @ 5059

Last change on this file since 5059 was 5059, checked in by cameron, 3 years ago

Pablo Kernel progress

File size: 5.9 KB
Line 
1#ifndef IDISA_BUILDER_H
2#define IDISA_BUILDER_H
3
4/*
5 *  Copyright (c) 2015 International Characters.
6 *  This software is licensed to the public under the Open Software License 3.0.
7 *  icgrep is a trademark of International Characters.
8 */
9#include <llvm/IR/Module.h>
10#include <llvm/IR/Constant.h>
11#include <llvm/IR/Type.h>
12#include <llvm/IR/Value.h>
13#include <llvm/IR/IRBuilder.h>
14
15using namespace llvm;
16
17namespace IDISA {
18
19class IDISA_Builder : public IRBuilder<> {
20public:
21
22    IDISA_Builder(Module * m, Type * bitBlockType)
23    : IRBuilder<>(m->getContext())
24    , mMod(m)
25    , mBitBlockType(bitBlockType)
26    , mBitBlockWidth(bitBlockType->isIntegerTy() ? cast<IntegerType>(bitBlockType)->getIntegerBitWidth() : cast<VectorType>(bitBlockType)->getBitWidth())
27    , mZeroInitializer(Constant::getNullValue(bitBlockType)) 
28    , mOneInitializer(Constant::getAllOnesValue(bitBlockType))
29    , mPrintRegisterFunction(nullptr) {
30
31    }
32
33    virtual ~IDISA_Builder() {}
34   
35    Type * getBitBlockType() const {
36        return mBitBlockType;
37    }
38
39    Value * bitCast(Value * a) {
40        return a->getType() == mBitBlockType ? a : CreateBitCast(a, mBitBlockType);
41    }
42
43    unsigned getBitBlockWidth() const {
44        return mBitBlockWidth;
45    }
46
47    Module * getModule() const {
48        return mMod;
49    }
50   
51    void setModule(Module * m)  {
52        mMod = m;
53    }
54   
55    Constant * allZeroes() const {
56        return mZeroInitializer;
57    }
58
59    Constant * allOnes() const {
60        return mOneInitializer;
61    }
62
63    LoadInst * CreateBlockAlignedLoad(Value * const ptr);
64    LoadInst * CreateBlockAlignedLoad(Value * const ptr, Value * const index);
65    LoadInst * CreateBlockAlignedLoad(Value * const ptr, std::initializer_list<Value *> indices);
66
67    void CreateBlockAlignedStore(Value * const value, Value * const ptr);
68    void CreateBlockAlignedStore(Value * const value, Value * const ptr, Value * const index);
69    void CreateBlockAlignedStore(Value * const value, Value * const ptr, std::initializer_list<Value *> indices);
70
71    void CallPrintRegister(const std::string & regName, Value * const value);
72    void CallPrintInt(const std::string & name, Value * const value);
73
74    Constant * simd_himask(unsigned fw);
75    Constant * simd_lomask(unsigned fw);
76
77    virtual Value * simd_add(unsigned fw, Value * a, Value * b);
78    virtual Value * simd_sub(unsigned fw, Value * a, Value * b);
79    virtual Value * simd_mult(unsigned fw, Value * a, Value * b);
80    virtual Value * simd_eq(unsigned fw, Value * a, Value * b);
81    virtual Value * simd_gt(unsigned fw, Value * a, Value * b);
82    virtual Value * simd_ugt(unsigned fw, Value * a, Value * b);
83    virtual Value * simd_lt(unsigned fw, Value * a, Value * b);
84    virtual Value * simd_ult(unsigned fw, Value * a, Value * b);
85    virtual Value * simd_max(unsigned fw, Value * a, Value * b);
86    virtual Value * simd_umax(unsigned fw, Value * a, Value * b);
87    virtual Value * simd_min(unsigned fw, Value * a, Value * b);
88    virtual Value * simd_umin(unsigned fw, Value * a, Value * b);
89    virtual Value * simd_if(unsigned fw, Value * cond, Value * a, Value * b);
90   
91    virtual Value * simd_slli(unsigned fw, Value * a, unsigned shift);
92    virtual Value * simd_srli(unsigned fw, Value * a, unsigned shift);
93    virtual Value * simd_srai(unsigned fw, Value * a, unsigned shift);
94   
95    virtual Value * simd_cttz(unsigned fw, Value * a);
96    virtual Value * simd_popcount(unsigned fw, Value * a);
97   
98    virtual Value * esimd_mergeh(unsigned fw, Value * a, Value * b);
99    virtual Value * esimd_mergel(unsigned fw, Value * a, Value * b);
100    virtual Value * esimd_bitspread(unsigned fw, Value * bitmask);
101   
102    virtual Value * hsimd_packh(unsigned fw, Value * a, Value * b);
103    virtual Value * hsimd_packl(unsigned fw, Value * a, Value * b);
104    virtual Value * hsimd_packh_in_lanes(unsigned lanes, unsigned fw, Value * a, Value * b);
105    virtual Value * hsimd_packl_in_lanes(unsigned lanes, unsigned fw, Value * a, Value * b);
106
107    virtual Value * hsimd_signmask(unsigned fw, Value * a);
108   
109    virtual Value * mvmd_extract(unsigned fw, Value * a, unsigned fieldIndex);
110    virtual Value * mvmd_insert(unsigned fw, Value * blk, Value * elt, unsigned fieldIndex);
111    virtual Value * mvmd_slli(unsigned fw, Value * a, unsigned shift);
112    virtual Value * mvmd_srli(unsigned fw, Value * a, unsigned shift);
113    virtual Value * mvmd_dslli(unsigned fw, Value * a, Value * b, unsigned shift);
114   
115    virtual Value * bitblock_any(Value * a);
116    Value * simd_and(Value * a, Value * b);
117    Value * simd_or(Value * a, Value * b);
118    Value * simd_xor(Value * a, Value * b);
119    Value * simd_not(Value * a);
120    Value * fwCast(unsigned fw, Value * a);
121   
122protected:
123    Module * mMod;
124    Type * mBitBlockType;
125    unsigned mBitBlockWidth;
126    Constant * mZeroInitializer;
127    Constant * mOneInitializer;
128    Constant * mPrintRegisterFunction;
129   
130    VectorType * fwVectorType(unsigned fw);
131};
132
133inline LoadInst * IDISA_Builder::CreateBlockAlignedLoad(Value * const ptr) {
134    return CreateAlignedLoad(ptr, mBitBlockWidth / 8);
135}
136
137inline LoadInst * IDISA_Builder::CreateBlockAlignedLoad(Value * const ptr, Value * const index) {
138    return CreateBlockAlignedLoad(CreateGEP(ptr, index));
139}
140
141inline LoadInst * IDISA_Builder::CreateBlockAlignedLoad(Value * const ptr, std::initializer_list<Value *> indicies) {
142    return CreateBlockAlignedLoad(CreateGEP(ptr, indicies));
143}
144
145inline void IDISA_Builder::CreateBlockAlignedStore(Value * const value, Value * const ptr) {
146    CreateAlignedStore(value, ptr, mBitBlockWidth / 8);
147}
148
149inline void IDISA_Builder::CreateBlockAlignedStore(Value * const value, Value * const ptr, Value * const index) {
150    CreateBlockAlignedStore(value, CreateGEP(ptr, index));
151}
152
153inline void IDISA_Builder::CreateBlockAlignedStore(Value * const value, Value * const ptr, std::initializer_list<Value *> indicies) {
154    CreateBlockAlignedStore(value, CreateGEP(ptr, indicies));
155}
156
157}
158#endif // IDISA_BUILDER_H
Note: See TracBrowser for help on using the repository browser.