diff --git a/mindspore/lite/src/scheduler.cc b/mindspore/lite/src/scheduler.cc index d316ed834cb..c7ebebd3c20 100644 --- a/mindspore/lite/src/scheduler.cc +++ b/mindspore/lite/src/scheduler.cc @@ -1467,7 +1467,7 @@ bool KernelFitCurrentSubGraph(const kernel::SubGraphType subgraph_type, const ke if (kernel.desc().arch != kernel::KERNEL_ARCH::kGPU) { return false; } - return (kernel.desc().data_type == kNumberTypeFloat16); + return (kernel.desc().data_type != kNumberTypeFloat32); case kernel::SubGraphType::kGpuFp32SubGraph: if (kernel.desc().arch != kernel::KERNEL_ARCH::kGPU) { return false;