Browse Source

fix non-arm82 ci

tags/20200916
nihui 5 years ago
parent
commit
71f86af8a6
1 changed files with 10 additions and 2 deletions
  1. +10
    -2
      src/net.cpp

+ 10
- 2
src/net.cpp View File

@@ -1126,7 +1126,8 @@ int Net::forward_layer(int layer_index, std::vector<Mat>& 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<Mat>& 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<Mat>& 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<Mat>& 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)
{


Loading…
Cancel
Save