fix bias stride

qsqqsqqsq-intellif · Dec 12, 2021 · 05b51c9 · 05b51c9
1 parent 7cf40e7
commit 05b51c9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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<ElementInputA*>(ptr_a), layout_A},\n");
   CutlassPrint(conv2d_decl, " {static_cast<ElementInputB*>(ptr_b), layout_B},\n");
   if (has_bias) {
-    CutlassPrint(conv2d_decl, " {static_cast<ElementOutput*>(ptr_c_bias), 0},\n");
+    CutlassPrint(
+        conv2d_decl,
+        " {static_cast<ElementOutput*>(ptr_c_bias), cutlass::layout::TensorNHWC::Stride(0)},\n");
   } else {
     CutlassPrint(conv2d_decl, " {static_cast<ElementOutput*>(ptr_out),layout_C},\n");
   }