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

[BACKEND][CODEGEN] Disabling L2 residency control by default (#570)

Browse Source
This commit is contained in:
Philippe Tillet
2022-06-29 17:05:13 -07:00
committed by GitHub
parent 1bbb2430d9
commit f733327ba4
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/codegen/selection/generator.cc
View File

@@ -802,6 +802,7 @@ void generator::visit_load_inst(ir::load_inst* x){
int width = std::min(tot_width, max_word_width); int width = std::min(tot_width, max_word_width);
int n_words = std::max(1, tot_width / width); int n_words = std::max(1, tot_width / width);
bool has_l2_evict_policy = (x->get_eviction_policy() != ir::load_inst::NORMAL) && tgt_->as_nvidia()->sm() >= 80; bool has_l2_evict_policy = (x->get_eviction_policy() != ir::load_inst::NORMAL) && tgt_->as_nvidia()->sm() >= 80;
has_l2_evict_policy = false;
// has_evict_policy = false; // currently disable until supported in `store` // has_evict_policy = false; // currently disable until supported in `store`
// ----- // -----
// create inline asm string // create inline asm string
@@ -966,6 +967,7 @@ void generator::visit_store_inst(ir::store_inst * x){
vec = std::min<size_t>(2, aln); vec = std::min<size_t>(2, aln);
} }
bool has_l2_evict_policy = (x->get_eviction_policy() != ir::load_inst::NORMAL) && tgt_->as_nvidia()->sm() >= 80; bool has_l2_evict_policy = (x->get_eviction_policy() != ir::load_inst::NORMAL) && tgt_->as_nvidia()->sm() >= 80;
has_l2_evict_policy = false;
auto idxs = idxs_.at(val_op); auto idxs = idxs_.at(val_op);
Type *ty = cvt(val_op->get_type()->get_scalar_ty()); Type *ty = cvt(val_op->get_type()->get_scalar_ty());
if (ty->isBFloatTy()) // llvm11-nvptx cannot select bf16 store if (ty->isBFloatTy()) // llvm11-nvptx cannot select bf16 store