Fix issue that argument buffer has limits.

liuliu · Nov 15, 2024 · dbdf28d · dbdf28d
1 parent 7832cde
commit dbdf28d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lib/nnc/cmd/ew/mps/ccv_nnc_ew_mps.m b/lib/nnc/cmd/ew/mps/ccv_nnc_ew_mps.m
@@ -28,6 +28,14 @@ static int _ccv_nnc_ewsum_forw(const ccv_nnc_cmd_t cmd, const ccv_nnc_hint_t hin
 			fallback_reason = "Disabled.";
 		}
 
+		if (use_mfa) {
+			if (input_size > 8) // Should be 31 or 16 maximum. We just be safe here.
+			{
+				use_mfa = false;
+				fallback_reason = "Too many argument buffers.";
+			}
+		}
+
 		if (use_mfa) {
 			for (z = 0; z < input_size; z++)
 			{