Skip to content

Commit

Permalink
Fix after rebase
Browse files Browse the repository at this point in the history
  • Loading branch information
jacobhinkle committed Jan 19, 2024
1 parent 7bfa709 commit fc07a9a
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions csrc/scheduler/mma_utils.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1289,9 +1289,10 @@ namespace {

void addMMAOp(Fusion* fusion_, std::vector<MulSumProperties>& props) {
for (auto prop : props) {
auto* init = IrBuilder::create<Val>(0.0, prop.out->getDataType().value());
auto* init =
IrBuilder::create<Val>(0.0, prop.insouts.out->getDataType().value());
IrBuilder::create<MmaOp>(
prop.insouts.out, prop.insouts..a, prop.insouts..b, init);
prop.insouts.out, prop.insouts.a, prop.insouts.b, init);
}
}

Expand Down

0 comments on commit fc07a9a

Please sign in to comment.