source: icGREP/icgrep-devel/icgrep/pablo/carry_manager.h @ 4689

Last change on this file since 4689 was 4689, checked in by cameron, 4 years ago

Ensure frame positions of nested blocks can be adjusted based on carry data size for the block.

File size: 3.7 KB
Line 
1/*
2 *  Copyright (c) 2015 International Characters.
3 *  This software is licensed to the public under the Open Software License 3.0.
4 *  icgrep is a trademark of International Characters.
5 */
6
7#ifndef CARRY_MANAGER_H
8#define CARRY_MANAGER_H
9#include <llvm/IR/IRBuilder.h>
10#include <llvm/IR/Module.h>
11#include <IDISA/idisa_builder.h>
12#include <pablo/codegenstate.h>
13#include <pablo/carry_data.h>
14
15/*
16 * Carry Data Manager.
17 *
18 * Each PabloBlock (Main, If, While) has a contiguous data area for carry information.
19 * The data area may be at a fixed or variable base offset from the base of the
20 * main function carry data area.
21 * The data area for each block consists of contiguous space for the local carries and
22 * advances of the block plus the areas of any ifs/whiles nested within the block.
23
24*/
25
26using namespace llvm;
27
28namespace pablo {
29
30class PabloBlock;
31
32
33
34class CarryManager {
35public:
36 
37    CarryManager(IRBuilder <> * b, VectorType * bitBlockType, ConstantAggregateZero * zero, Constant * one, IDISA::IDISA_Builder * idb)
38    : mBuilder(b)
39    , mBitBlockType(bitBlockType)
40    , mZeroInitializer(zero)
41    , mOneInitializer(one)
42    , iBuilder(idb)
43    , mPabloRoot(nullptr)
44    , mCurrentScope(nullptr)
45    , mCarryInfo(nullptr)
46    , mCurrentScopeIndex(0)
47    , mCarryDataPtr(nullptr)
48    , mBlockNoPtr(nullptr)
49    , mBlockNo(nullptr)
50    , mTotalCarryDataBitBlocks(0)
51    {
52
53    }
54   
55    unsigned initialize(PabloBlock * blk, Value * carryDataPtr); 
56   
57    unsigned enumerate(PabloBlock * blk, unsigned ifDepth, unsigned whileDepth);
58   
59    void generateBlockNoIncrement();
60   
61    Value * getBlockNoPtr();
62   
63    /* Entering and leaving scopes. */
64   
65    void enterScope(PabloBlock * blk);
66
67    void leaveScope();
68   
69    /* Methods for processing individual carry-generating operations. */
70   
71    Value * getCarryOpCarryIn(int localIndex);
72
73    void setCarryOpCarryOut(unsigned idx, Value * carry_out);
74
75    Value * advanceCarryInCarryOut(int localIndex, int shift_amount, Value * strm);
76 
77    /* Methods for getting and setting carry summary values for If statements */
78   
79    bool blockHasCarries();
80   
81    Value * getCarrySummaryExpr();
82   
83    void generateCarryOutSummaryCodeIfNeeded();
84   
85    void addSummaryPhiIfNeeded(BasicBlock * ifEntryBlock, BasicBlock * ifBodyFinalBlock);
86   
87    /* Methods for load/store of carries for non-while blocks. */
88   
89    void ensureCarriesLoadedLocal();
90
91    void ensureCarriesStoredLocal();
92   
93    /* Methods for handling while statements */
94   
95    void ensureCarriesLoadedRecursive();
96
97    void initializeCarryDataPhisAtWhileEntry(BasicBlock * whileBodyFinalBlock);
98
99    void extendCarryDataPhisAtWhileBodyFinalBlock(BasicBlock * whileBodyFinalBlock);
100
101    void ensureCarriesStoredRecursive();
102
103   
104private:
105    IRBuilder <> * mBuilder;
106    VectorType * mBitBlockType;
107    ConstantAggregateZero * mZeroInitializer;
108    Constant * mOneInitializer;
109    IDISA::IDISA_Builder * iBuilder;
110    PabloBlock * mPabloRoot;
111    PabloBlock * mCurrentScope;
112    PabloBlockCarryData * mCarryInfo;
113    unsigned mCurrentScopeIndex;
114    Value * mCarryDataPtr;
115    Value * mBlockNoPtr;
116    Value * mBlockNo;
117    unsigned mTotalCarryDataBitBlocks;
118   
119    std::vector<PabloBlockCarryData *> mCarryInfoVector;
120
121
122    std::vector<Value *> mCarryInVector;
123    std::vector<PHINode *> mCarryInPhis; 
124    std::vector<PHINode *> mCarryOutAccumPhis; 
125    std::vector<Value *> mCarryOutVector;
126
127    Value * unitAdvanceCarryInCarryOut(int localIndex, Value * strm);
128    Value * shortAdvanceCarryInCarryOut(int localIndex, int shift_amount, Value * strm);
129    Value * longAdvanceCarryInCarryOut(int localIndex, int shift_amount, Value * strm);
130   
131};
132
133}
134
135#endif // CARRY_MANAGER_H
Note: See TracBrowser for help on using the repository browser.