gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[code generation] some more cleaning

Browse Source
This commit is contained in:
Philippe Tillet
2019-01-08 12:39:25 -05:00
parent 297d1a99d1
commit 58757b8f10
7 changed files with 153 additions and 458 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
include/codegen/allocation.h
View File

@@ -21,17 +21,14 @@ class loop_info;
class allocation {
public:
// accessors
unsigned get_num_bytes(ir::value *x) const;
unsigned get_offset(ir::value *x) const { return offsets_.at(x); }
unsigned get_allocated_size() const { return allocated_size_; }
bool has_double_buffer(ir::value *x) const { return double_buffer_.find(x) != double_buffer_.end(); }
// run
void run(ir::function &fn);
private:
std::map<ir::value*, unsigned> offsets_;
std::set<ir::value*> double_buffer_;
std::map<ir::value*, unsigned> num_bytes_;
size_t allocated_size_;
// dependences