Skip to content

Commit

Permalink
Updated tir::PrimFunc usage in cmsisnn contrib
Browse files Browse the repository at this point in the history
  • Loading branch information
Lunderberg committed Apr 19, 2022
1 parent 0d95f1c commit f1579c7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/relay/backend/contrib/cmsisnn/relay_to_tir.cc
Original file line number Diff line number Diff line change
Expand Up @@ -108,7 +108,7 @@ class RelayToTIRVisitor : public MixedModeMutator {
}

tir::PrimFunc replacement_func(func_signature, body, VoidType(), buffer_map,
Map<tir::Var, tir::Buffer>(), DictAttrs(dict_attrs));
DictAttrs(dict_attrs));
ir_module_->Add(global_var, replacement_func);
}

Expand Down

0 comments on commit f1579c7

Please sign in to comment.