Skip to content

GitLab

Unverified Commit e547d247 authored by Lei Wang's avatar Lei Wang Committed by GitHub
Browse files

[Bugfix] Update TIR registration for GemmSPPy to use tile operation (#1361)

parent 283a9a00
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/op/gemm_sp_py.cc
View file @ e547d247
...@@ -279,7 +279,7 @@ LayoutMap GemmSPPyNode::InferLayout(const LayoutInferArgs &T, ...@@ -279,7 +279,7 @@ LayoutMap GemmSPPyNode::InferLayout(const LayoutInferArgs &T,
return results; return results;
} }
TIR_REGISTER_TL_OP(GemmSPPy, gemm_sp_py) TIR_REGISTER_TL_TILE_OP(GemmSPPy, gemm_sp_py)
.set_num_inputs(5) .set_num_inputs(5)
.set_attr<TCallEffectKind>("TCallEffectKind", .set_attr<TCallEffectKind>("TCallEffectKind",
Integer(CallEffectKind::kOpaque)); Integer(CallEffectKind::kOpaque));
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment