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

[BACKEND][CODEGEN] Faster reduction for scanline layout (#516)

Browse Source
This commit is contained in:
Philippe Tillet
2022-05-14 15:26:13 -07:00
committed by GitHub
parent d1a22a94e6
commit d35617bea1
3 changed files with 65 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
include/triton/codegen/selection/generator.h
View File

@@ -177,6 +177,7 @@ public:
void visit_sqrt_inst(ir::sqrt_inst*);
Value* shfl_sync(Value* acc, int32_t i);
void visit_reduce1d_inst(ir::reduce_inst*, std::function<Value*(Value*,Value*)>, Value*);
void visit_reducend_inst_fast(ir::reduce_inst* x, std::function<Value*(Value*,Value*)> do_acc, Value *neutral);
void visit_reducend_inst(ir::reduce_inst*, std::function<Value*(Value*,Value*)>, Value*);
void visit_reduce_inst(ir::reduce_inst*);
void visit_select_inst(ir::select_inst*);