This commit is contained in:
gongdaguo 2022-02-15 10:16:24 +08:00
parent 9fe6b148cc
commit 786cee1e2e
2 changed files with 4 additions and 1 deletions

View File

@ -278,6 +278,9 @@ endif()
if(MSVC)
set(MSLITE_ENABLE_CONVERTER off)
endif()
if(MSLITE_ENABLE_CONVERTER)
add_compile_definitions(ENABLE_CONVERTER)
endif()
message(STATUS "************MindSpore Lite Build Option:************")
message(STATUS "\tMSLITE_GPU_BACKEND = \t${MSLITE_GPU_BACKEND}")

View File

@ -209,7 +209,7 @@ REG_KERNEL(kCPU, kNumberTypeInt8, PrimitiveType_Cast, LiteKernelCreator<CastCPUK
REG_KERNEL(kCPU, kNumberTypeInt32, PrimitiveType_Cast, LiteKernelCreator<CastCPUKernel>)
REG_KERNEL(kCPU, kNumberTypeInt64, PrimitiveType_Cast, LiteKernelCreator<CastCPUKernel>)
REG_KERNEL(kCPU, kNumberTypeBool, PrimitiveType_Cast, LiteKernelCreator<CastCPUKernel>)
#ifndef ENABLE_ARM
#ifdef ENABLE_CONVERTER
REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Cast, LiteKernelCreator<CastCPUKernel>)
#endif
} // namespace mindspore::kernel