From e60315222057b234c02132fde12ccd5f2b079caa Mon Sep 17 00:00:00 2001 From: cjh9368 Date: Sat, 31 Oct 2020 16:53:12 +0800 Subject: [PATCH] fp16 to fp32 support only x86_64 platform --- mindspore/lite/src/runtime/kernel/arm/fp32/cast.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/lite/src/runtime/kernel/arm/fp32/cast.cc b/mindspore/lite/src/runtime/kernel/arm/fp32/cast.cc index cf07155ea1..22a044c927 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp32/cast.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp32/cast.cc @@ -166,7 +166,7 @@ REG_KERNEL(kCPU, kNumberTypeFloat32, PrimitiveType_Cast, CpuCastFp32KernelCreato REG_KERNEL(kCPU, kNumberTypeUInt8, PrimitiveType_Cast, CpuCastFp32KernelCreator) REG_KERNEL(kCPU, kNumberTypeInt8, PrimitiveType_Cast, CpuCastFp32KernelCreator) REG_KERNEL(kCPU, kNumberTypeInt32, PrimitiveType_Cast, CpuCastFp32KernelCreator) -#ifndef ENABLE_ARM64 +#ifndef ENABLE_ARM REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Cast, CpuCastFp32KernelCreator) #endif } // namespace mindspore::kernel