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

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

simd_himask

File size: 3.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 {
20public:
21
22    IDISA_Builder(Type * bitBlockType)
23    : mMod(nullptr)
24    , mLLVMBuilder(nullptr)
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    virtual ~IDISA_Builder() {};
33
34    void initialize(Module * m, IRBuilder <> * b) {
35        mMod = m;
36        mLLVMBuilder = b;
37    }
38   
39    Type * getBitBlockType() { return mBitBlockType;}
40    Value * bitCast(Value * a) {return a->getType() == mBitBlockType ? a : mLLVMBuilder->CreateBitCast(a, mBitBlockType);}
41    int getBitBlockWidth() { return mBitBlockWidth;}
42    Module * getModule() {return mMod;}
43    void genPrintRegister(std::string regName, Value * bitblockValue);
44   
45   
46    Constant * allZeroes() {return mZeroInitializer;}
47    Constant * allOnes() {return mOneInitializer;}
48    Constant * simd_himask(unsigned fw);
49       
50    virtual Value * simd_add(unsigned fw, Value * a, Value * b);
51    virtual Value * simd_sub(unsigned fw, Value * a, Value * b);
52    virtual Value * simd_mult(unsigned fw, Value * a, Value * b);
53    virtual Value * simd_eq(unsigned fw, Value * a, Value * b);
54    virtual Value * simd_gt(unsigned fw, Value * a, Value * b);
55    virtual Value * simd_ugt(unsigned fw, Value * a, Value * b);
56    virtual Value * simd_lt(unsigned fw, Value * a, Value * b);
57    virtual Value * simd_ult(unsigned fw, Value * a, Value * b);
58    virtual Value * simd_max(unsigned fw, Value * a, Value * b);
59    virtual Value * simd_umax(unsigned fw, Value * a, Value * b);
60    virtual Value * simd_min(unsigned fw, Value * a, Value * b);
61    virtual Value * simd_umin(unsigned fw, Value * a, Value * b);
62    virtual Value * simd_if(unsigned fw, Value * cond, Value * a, Value * b);
63   
64    virtual Value * simd_slli(unsigned fw, Value * a, unsigned shift);
65    virtual Value * simd_srli(unsigned fw, Value * a, unsigned shift);
66    virtual Value * simd_srai(unsigned fw, Value * a, unsigned shift);
67   
68    virtual Value * simd_cttz(unsigned fw, Value * a);
69    virtual Value * simd_popcount(unsigned fw, Value * a);
70   
71    virtual Value * esimd_mergeh(unsigned fw, Value * a, Value * b);
72    virtual Value * esimd_mergel(unsigned fw, Value * a, Value * b);
73    virtual Value * esimd_bitspread(unsigned fw, Value * bitmask);
74   
75    virtual Value * hsimd_packh(unsigned fw, Value * a, Value * b);
76    virtual Value * hsimd_packl(unsigned fw, Value * a, Value * b);
77    virtual Value * hsimd_signmask(unsigned fw, Value * a);
78
79   
80    virtual Value * mvmd_extract(unsigned fw, Value * a, unsigned fieldIndex);
81    virtual Value * mvmd_insert(unsigned fw, Value * blk, Value * elt, unsigned fieldIndex);
82    virtual Value * mvmd_dslli(unsigned fw, Value * a, Value * b, unsigned shift);
83   
84    virtual Value * bitblock_any(Value * a);
85    Value * simd_and(Value * a, Value * b);
86    Value * simd_or(Value * a, Value * b);
87    Value * simd_xor(Value * a, Value * b);
88    Value * simd_not(Value * a);
89    Value * fwCast(unsigned fw, Value * a);
90   
91protected:
92    Module * mMod;
93    IRBuilder <> * mLLVMBuilder;
94    Type * mBitBlockType;
95    unsigned mBitBlockWidth;
96    Constant * mZeroInitializer;
97    Constant * mOneInitializer;
98    Constant * mPrintRegisterFunction;
99   
100    VectorType * fwVectorType(unsigned fw);
101};
102
103}
104#endif // IDISA_BUILDER_H
Note: See TracBrowser for help on using the repository browser.