Merge pull request #2804 from alibaba/feature/metalbugfix

Fix bug for metal run for fp16 model
alibaba · Mar 26, 2024 · 2d926e4 · 2d926e4
2 parents ffde3f4 + 7f65266
commit 2d926e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/backend/metal/MetalConvolutionCommon.mm b/source/backend/metal/MetalConvolutionCommon.mm
@@ -153,7 +153,7 @@ void weightInBlock(int group, int oc, int ic, int kh, int kw, const FType *src,
     auto ic     = size / kw / kh / (oc / group);
 
     // convert
-    if (loadWeightInt8) {
+    if (loadWeightInt8 && qnt->weight.get() != nullptr) {
         auto backend = static_cast<MetalBackend *>(this->backend());
         mWeight = weightTransform(group, oc, ic, kh, kw, (float*)qnt->weight.get(), !qnt->canUseInt4, qnt->canUseInt4);
         auto dequantParams = getDequantScale(qnt->alpha.get(), qnt->alpha.size(), backend, qnt->asymmetric);