update ConvKernel isAvailable checking

MegEngine · Oct 29, 2023 · 08f1655 · 08f1655
1 parent f18ba39
commit 08f1655
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/compiler/lib/KernelGen/Common/ConvKernel.h b/compiler/lib/KernelGen/Common/ConvKernel.h
@@ -14,7 +14,7 @@ class ConvImpl : public KernelFunc {
     static bool is_channel_broadcast_bias(TContext* ctx) {
         if (is_bias(ctx)) {
             CCOperand bias = ctx->getAttrOprand("operand:2");
-            return bias.shape[0] == 1 && bias.shape[2] == 1 && bias.shape[3] == 1;
+            return (bias.shape[0] == 1 && bias.shape[2] == 1 && bias.shape[3] == 1) || bias.shape.size() == 1;
         }
         return false;
     }