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

some cleaning

Browse Source
This commit is contained in:
Philippe Tillet
2019-06-24 09:31:34 -07:00
parent 67989e7d18
commit f257884eb7
7 changed files with 71 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
lib/codegen/shmem_allocation.cpp
View File

@@ -12,12 +12,6 @@ namespace triton{
namespace codegen{
unsigned shmem_allocation::is_ld_padded(ir::value *x) {
if(auto* phi = dynamic_cast<ir::phi_node*>(x)) {
unsigned result = 0;
for(unsigned i = 0; i < phi->get_num_incoming(); i++)
result = std::max(result, is_ld_padded(phi->get_incoming_value(i)));
return result;
}
if(dynamic_cast<ir::trans_inst*>(x))
return 4;
for(ir::user* user: x->get_users())
@@ -25,7 +19,13 @@ unsigned shmem_allocation::is_ld_padded(ir::value *x) {
if(params_->get_fragment(user, 0) == tune::HMMA_FRAGMENT_C){
return 16;
}
return 16;
if(auto* phi = dynamic_cast<ir::phi_node*>(x)) {
unsigned result = 0;
for(unsigned i = 0; i < phi->get_num_incoming(); i++)
result = std::max(result, is_ld_padded(phi->get_incoming_value(i)));
return result;
}
return 0;
}
unsigned shmem_allocation::get_num_bytes(ir::value *x) {