diff --git a/src/relay/backend/contrib/cutlass/codegen.cc b/src/relay/backend/contrib/cutlass/codegen.cc index 79b63a8a9147..bc4e6905a162 100644 --- a/src/relay/backend/contrib/cutlass/codegen.cc +++ b/src/relay/backend/contrib/cutlass/codegen.cc @@ -331,7 +331,9 @@ std::string Conv2dOp(std::string id, const Str2StrMap& attrs, CutlassPrint(conv2d_decl, " {static_cast(ptr_a), layout_A},\n"); CutlassPrint(conv2d_decl, " {static_cast(ptr_b), layout_B},\n"); if (has_bias) { - CutlassPrint(conv2d_decl, " {static_cast(ptr_c_bias), 0},\n"); + CutlassPrint( + conv2d_decl, + " {static_cast(ptr_c_bias), cutlass::layout::TensorNHWC::Stride(0)},\n"); } else { CutlassPrint(conv2d_decl, " {static_cast(ptr_out),layout_C},\n"); }