cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: TFLiteGPURunner crash on Linux when use_advanced_gpu_api is true

Browse Source
This commit is contained in:
homuler 2021-08-26 12:21:16 +09:00
parent 1faeaae7e5
commit 2eb19b739d
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mediapipe/util/tflite/tflite_gpu_runner.cc
View File

@ -252,8 +252,10 @@ absl::Status TFLiteGPURunner::InitializeOpenCL(
MP_RETURN_IF_ERROR(cl_environment_->NewInferenceBuilder( MP_RETURN_IF_ERROR(cl_environment_->NewInferenceBuilder(
cl_options, std::move(graph_cl), builder)); cl_options, std::move(graph_cl), builder));
#endif // __ANDROID__
return absl::OkStatus(); return absl::OkStatus();
#else
return mediapipe::UnimplementedError("Currently only Android is supported");
#endif // __ANDROID__
} }
#ifdef __ANDROID__ #ifdef __ANDROID__