Skip to content

Commit

Permalink
wordaround wenet 2
Browse files Browse the repository at this point in the history
Change-Id: I8d7eac0d2e445f2eae9a6d2dab18c3b68a016921
  • Loading branch information
sailist committed Nov 5, 2024
1 parent d19605d commit f732432
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion lib/Dialect/Tpu/Transforms/LayerGroup/LmemAllocator.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1369,7 +1369,7 @@ void LmemAllocator::sc_method_multi_core_v3(
<< "; cost = " << _group_cost
<< "\n";
});
if (_group_cost == 96230) {
if (_group_cost == 88611) {
for (auto op : lg_info.group_ops) {
if (auto matmul = dyn_cast<tpu::MatMulOp>(op)) {
if (matmul.getHdimIsBatch()) {
Expand Down

0 comments on commit f732432

Please sign in to comment.