forked from mindspore-Ecosystem/mindspore
fix build java error
This commit is contained in:
parent
86bbd1dc98
commit
7587a5b7b3
|
@ -304,6 +304,7 @@ checkopts()
|
|||
LITE_LANGUAGE="cpp"
|
||||
elif [[ "$OPTARG" == "java" ]]; then
|
||||
LITE_LANGUAGE="java"
|
||||
ENABLE_CONVERTER="off"
|
||||
elif [[ "$OPTARG" == "object-c" ]]; then
|
||||
LITE_LANGUAGE="object-c"
|
||||
else
|
||||
|
|
|
@ -116,8 +116,11 @@ kernel::LiteKernel *KernelRegistry::GetKernel(const std::vector<Tensor *> &in_te
|
|||
auto creator = GetCreator(key);
|
||||
if (creator != nullptr) {
|
||||
auto kernel = creator(in_tensors, out_tensors, parameter, ctx, key, primitive);
|
||||
return kernel;
|
||||
if (kernel != nullptr) {
|
||||
return kernel;
|
||||
}
|
||||
}
|
||||
free(parameter);
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue