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; }