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

[CODEGEN] Various bugfixes that make it possible to fuse RNG in a matmul epilogue (#356)

Browse Source
This commit is contained in:
Philippe Tillet
2021-10-24 02:30:46 -07:00
committed by GitHub
parent 858dec8372
commit 5ce1b726dc
17 changed files with 149 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lib/codegen/analysis/axes.cc
View File

@@ -116,7 +116,8 @@ void axes::update_graph(ir::instruction *i) {
switch (i->get_id()) {
case ir::INST_REDUCE: return update_graph_reduce(i);
case ir::INST_RESHAPE: return update_graph_reshape(i);
case ir::INST_SPLAT: return update_graph_no_edge(i);;
case ir::INST_SPLAT: return update_graph_no_edge(i);
case ir::INST_CAT: return update_graph_elementwise(i, true);
case ir::INST_TRANS: return update_graph_trans(i);
case ir::INST_BROADCAST: return update_graph_broadcast(i);
case ir::INST_DOT: return update_graph_dot(i);

1
lib/codegen/analysis/layout.cc
View File

@@ -499,6 +499,7 @@ void layouts::run(ir::module &mod) {
make_graph(i);
});
// connected components
graph_.connected_components(&values_, &groups_);