From b2e8a8d87dfb9fe3e312374a22ca316bb780fbe7 Mon Sep 17 00:00:00 2001 From: nihui Date: Wed, 30 Jun 2021 23:57:16 +0800 Subject: [PATCH] split noop support riscv zfh --- src/layer/noop.cpp | 2 +- src/layer/split.cpp | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/layer/noop.cpp b/src/layer/noop.cpp index 36e0f7b09..68572b0ba 100644 --- a/src/layer/noop.cpp +++ b/src/layer/noop.cpp @@ -22,7 +22,7 @@ Noop::Noop() support_inplace = true; support_vulkan = true; support_packing = true; - support_fp16_storage = cpu_support_arm_asimdhp(); + support_fp16_storage = cpu_support_arm_asimdhp() || cpu_support_riscv_zfh(); support_bf16_storage = true; support_image_storage = true; } diff --git a/src/layer/split.cpp b/src/layer/split.cpp index d517dec40..f79fce0f1 100644 --- a/src/layer/split.cpp +++ b/src/layer/split.cpp @@ -23,7 +23,7 @@ Split::Split() support_inplace = false; support_vulkan = true; support_packing = true; - support_fp16_storage = cpu_support_arm_asimdhp(); + support_fp16_storage = cpu_support_arm_asimdhp() || cpu_support_riscv_zfh(); support_bf16_storage = true; support_image_storage = true; }