diff --git a/src/op/gemm_sp_py.cc b/src/op/gemm_sp_py.cc index d7c4f4aab..6ad8ca9b5 100644 --- a/src/op/gemm_sp_py.cc +++ b/src/op/gemm_sp_py.cc @@ -279,7 +279,7 @@ LayoutMap GemmSPPyNode::InferLayout(const LayoutInferArgs &T, return results; } -TIR_REGISTER_TL_OP(GemmSPPy, gemm_sp_py) +TIR_REGISTER_TL_TILE_OP(GemmSPPy, gemm_sp_py) .set_num_inputs(5) .set_attr("TCallEffectKind", Integer(CallEffectKind::kOpaque));