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

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

Parallel long addition within icgrep improves performance on AVX2

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