source: icGREP/icgrep-devel/icgrep/IR_Gen/idisa_builder.h @ 5240

Last change on this file since 5240 was 5240, checked in by nmedfort, 3 years ago

Cleaned up memory leaks + some warning messages.

File size: 6.8 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#include <llvm/Support/Host.h>
15#include <llvm/ADT/Triple.h>
16#include <IR_Gen/CBuilder.h>
17#include <IR_Gen/types/streamtype.h>
18#include <boost/container/flat_map.hpp>
19
20using namespace llvm;
21
22namespace IDISA {
23
24class IDISA_Builder : public CBuilder {
25
26    using StreamTypes = boost::container::flat_map<unsigned, StreamType *>;
27
28public:
29
30    IDISA_Builder(Module * m, unsigned archBitWidth, unsigned bitBlockWidth, unsigned stride, unsigned CacheAlignment=64);
31
32    virtual ~IDISA_Builder();
33   
34    std::string getBitBlockTypeName() const;  // A short string such as v4i64 or i256.
35
36    Value * bitCast(Value * a) {
37        return (a->getType() == mBitBlockType) ? a : CreateBitCast(a, mBitBlockType);
38    }
39
40    unsigned getBitBlockWidth() const {
41        return mBitBlockWidth;
42    }
43
44    unsigned getStride() const {
45        return mStride;
46    }
47
48    Constant * allZeroes() const {
49        return mZeroInitializer;
50    }
51
52    Constant * allOnes() const {
53        return mOneInitializer;
54    }
55   
56
57    LoadInst * CreateBlockAlignedLoad(Value * const ptr);
58    LoadInst * CreateBlockAlignedLoad(Value * const ptr, Value * const index);
59    LoadInst * CreateBlockAlignedLoad(Value * const ptr, std::initializer_list<Value *> indices);
60
61    void CreateBlockAlignedStore(Value * const value, Value * const ptr);
62    void CreateBlockAlignedStore(Value * const value, Value * const ptr, Value * const index);
63    void CreateBlockAlignedStore(Value * const value, Value * const ptr, std::initializer_list<Value *> indices);
64
65    void CallPrintRegister(const std::string & regName, Value * const value);
66   
67    VectorType * fwVectorType(unsigned fw);
68
69    Constant * simd_himask(unsigned fw);
70    Constant * simd_lomask(unsigned fw);
71   
72    virtual Value * simd_fill(unsigned fw, Value * a);
73
74    virtual Value * simd_add(unsigned fw, Value * a, Value * b);
75    virtual Value * simd_sub(unsigned fw, Value * a, Value * b);
76    virtual Value * simd_mult(unsigned fw, Value * a, Value * b);
77    virtual Value * simd_eq(unsigned fw, Value * a, Value * b);
78    virtual Value * simd_gt(unsigned fw, Value * a, Value * b);
79    virtual Value * simd_ugt(unsigned fw, Value * a, Value * b);
80    virtual Value * simd_lt(unsigned fw, Value * a, Value * b);
81    virtual Value * simd_ult(unsigned fw, Value * a, Value * b);
82    virtual Value * simd_max(unsigned fw, Value * a, Value * b);
83    virtual Value * simd_umax(unsigned fw, Value * a, Value * b);
84    virtual Value * simd_min(unsigned fw, Value * a, Value * b);
85    virtual Value * simd_umin(unsigned fw, Value * a, Value * b);
86    virtual Value * simd_if(unsigned fw, Value * cond, Value * a, Value * b);
87   
88    virtual Value * simd_slli(unsigned fw, Value * a, unsigned shift);
89    virtual Value * simd_srli(unsigned fw, Value * a, unsigned shift);
90    virtual Value * simd_srai(unsigned fw, Value * a, unsigned shift);
91   
92    virtual Value * simd_cttz(unsigned fw, Value * a);
93    virtual Value * simd_popcount(unsigned fw, Value * a);
94   
95    virtual Value * esimd_mergeh(unsigned fw, Value * a, Value * b);
96    virtual Value * esimd_mergel(unsigned fw, Value * a, Value * b);
97    virtual Value * esimd_bitspread(unsigned fw, Value * bitmask);
98   
99    virtual Value * hsimd_packh(unsigned fw, Value * a, Value * b);
100    virtual Value * hsimd_packl(unsigned fw, Value * a, Value * b);
101    virtual Value * hsimd_packh_in_lanes(unsigned lanes, unsigned fw, Value * a, Value * b);
102    virtual Value * hsimd_packl_in_lanes(unsigned lanes, unsigned fw, Value * a, Value * b);
103
104    virtual Value * hsimd_signmask(unsigned fw, Value * a);
105   
106    virtual Value * mvmd_extract(unsigned fw, Value * a, unsigned fieldIndex);
107    virtual Value * mvmd_insert(unsigned fw, Value * blk, Value * elt, unsigned fieldIndex);
108    virtual Value * mvmd_slli(unsigned fw, Value * a, unsigned shift);
109    virtual Value * mvmd_srli(unsigned fw, Value * a, unsigned shift);
110    virtual Value * mvmd_dslli(unsigned fw, Value * a, Value * b, unsigned shift);
111   
112   
113    virtual Value * bitblock_any(Value * a);
114    // full add producing {carryout, sum}
115    virtual std::pair<Value *, Value *> bitblock_add_with_carry(Value * a, Value * b, Value * carryin);
116    // full shift producing {shiftout, shifted}
117    virtual std::pair<Value *, Value *> bitblock_advance(Value * a, Value * shiftin, unsigned shift);
118    virtual Value * bitblock_mask_from(Value * pos);
119    virtual Value * bitblock_set_bit(Value * pos);
120   
121    Value * simd_and(Value * a, Value * b);
122    Value * simd_or(Value * a, Value * b);
123    Value * simd_xor(Value * a, Value * b);
124    Value * simd_not(Value * a);
125    Value * fwCast(unsigned fw, Value * a);
126   
127    inline VectorType * getBitBlockType() const {
128        return mBitBlockType;
129    }
130
131    inline Type * getStreamSetTy(const unsigned NumElements = 1, const unsigned FieldWidth = 1) {
132        return ArrayType::get(getStreamTy(FieldWidth), NumElements);
133    }
134   
135    Type * getStreamTy(const unsigned FieldWidth = 1);
136
137    inline llvm::AllocaInst * CreateCacheAlignedAlloca(llvm::Type * Ty, llvm::Value * ArraySize = nullptr) {
138        llvm::AllocaInst * instr = CreateAlloca(Ty, ArraySize);
139        instr->setAlignment(mCacheLineAlignment);
140        return instr;
141    }
142   
143protected:
144    unsigned            mBitBlockWidth;
145    unsigned            mStride;
146    VectorType *        mBitBlockType;
147
148    Constant *          mZeroInitializer;
149    Constant *          mOneInitializer;
150    Constant *          mPrintRegisterFunction;
151    StreamTypes         mStreamTypes;
152};
153
154inline LoadInst * IDISA_Builder::CreateBlockAlignedLoad(Value * const ptr) {
155    return CreateAlignedLoad(ptr, mBitBlockWidth / 8);
156}
157
158inline LoadInst * IDISA_Builder::CreateBlockAlignedLoad(Value * const ptr, Value * const index) {
159    return CreateBlockAlignedLoad(CreateGEP(ptr, index));
160}
161
162inline LoadInst * IDISA_Builder::CreateBlockAlignedLoad(Value * const ptr, std::initializer_list<Value *> indices) {
163    return CreateBlockAlignedLoad(CreateGEP(ptr, indices));
164}
165
166inline void IDISA_Builder::CreateBlockAlignedStore(Value * const value, Value * const ptr) {
167    CreateAlignedStore(value, ptr, mBitBlockWidth / 8);
168}
169
170inline void IDISA_Builder::CreateBlockAlignedStore(Value * const value, Value * const ptr, Value * const index) {
171    CreateBlockAlignedStore(value, CreateGEP(ptr, index));
172}
173
174inline void IDISA_Builder::CreateBlockAlignedStore(Value * const value, Value * const ptr, std::initializer_list<Value *> indices) {
175    CreateBlockAlignedStore(value, CreateGEP(ptr, indices));
176}
177   
178
179   
180}
181#endif // IDISA_BUILDER_H
Note: See TracBrowser for help on using the repository browser.