From 71f86af8a680f910c2ecc067aa5e772e1db47f6c Mon Sep 17 00:00:00 2001 From: nihui Date: Tue, 28 Jul 2020 22:51:29 +0800 Subject: [PATCH] fix non-arm82 ci --- src/net.cpp | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/src/net.cpp b/src/net.cpp index 2c6717a26..98b1e63ae 100644 --- a/src/net.cpp +++ b/src/net.cpp @@ -1126,7 +1126,8 @@ int Net::forward_layer(int layer_index, std::vector& blob_mats, const Optio } } - if (opt.use_fp16_storage) +#if NCNN_ARM82 + if (opt.use_fp16_storage && cpu_support_arm_asimdhp()) { if (bottom_blob.elemsize / bottom_blob.elempack == 4u && layer->support_fp16_storage) { @@ -1142,6 +1143,9 @@ int Net::forward_layer(int layer_index, std::vector& blob_mats, const Optio } } else if (opt.use_bf16_storage) +#else + if (opt.use_bf16_storage) +#endif { if (bottom_blob.elemsize / bottom_blob.elempack == 4u && layer->support_bf16_storage) { @@ -1240,7 +1244,8 @@ int Net::forward_layer(int layer_index, std::vector& blob_mats, const Optio } } - if (opt.use_fp16_storage) +#if NCNN_ARM82 + if (opt.use_fp16_storage && cpu_support_arm_asimdhp()) { if (bottom_blobs[i].elemsize / bottom_blobs[i].elempack == 4u && layer->support_fp16_storage) { @@ -1256,6 +1261,9 @@ int Net::forward_layer(int layer_index, std::vector& blob_mats, const Optio } } else if (opt.use_bf16_storage) +#else + if (opt.use_bf16_storage) +#endif { if (bottom_blobs[i].elemsize / bottom_blobs[i].elempack == 4u && layer->support_bf16_storage) {