Add more error checking

Tencent · Sep 5, 2023 · f489d44 · f489d44
1 parent a273057
commit f489d44
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/layer/linearint8.cpp b/src/layer/linearint8.cpp
@@ -27,6 +27,8 @@ int LinearInt8::load_param(const ParamDict& pd)
     in_dim = pd.get(0, 0);
     out_dim = pd.get(1, 0);
     group_size = pd.get(2, 1);
+    if (in_dim * out_dim % group_size)
+        return -1;
     return 0;
 }