fix(compiler)aadd ok bias flag on available

MegEngine · Oct 23, 2024 · 45f6862 · 45f6862
1 parent 45faa53
commit 45f6862
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/compiler/lib/KernelGen/WebAssembly/MatMulKernel/FP32MatMulM4N12.cpp b/compiler/lib/KernelGen/WebAssembly/MatMulKernel/FP32MatMulM4N12.cpp
@@ -29,7 +29,8 @@ bool Fp32MatMulM4N12::IsAvailable(TContext* context) const {
                     context->getAttrOprand("operand:0").shape.size() == 2;
     bool ok_tran = context->getAttrBool("transposeA") == false &&
                    context->getAttrBool("transposeB") == false;
-    return ok_dtype && ok_mode && ok_shape && ok_tran;    
+    bool ok_bias = context->getAttrInt("nr_operands") == 3;
+    return ok_dtype && ok_mode && ok_shape && ok_tran && ok_bias;    
 }