talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Verify that kernel cache is only used when OpenCL is active

Browse Source 
PiperOrigin-RevId: 491463306
This commit is contained in:
MediaPipe Team
2022-11-28 15:47:55 -08:00
committed by Copybara-Service
parent 26a7ca5c64
commit 7b74fd53f5
3 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/util/tflite/tflite_gpu_runner.h
+3 -1
View File
@@ -21,6 +21,7 @@
#include "absl/status/status.h"
#include "mediapipe/framework/port.h"
#include "mediapipe/framework/port/ret_check.h"
#include "mediapipe/framework/port/status.h"
#include "mediapipe/framework/port/statusor.h"
#include "tensorflow/lite/core/api/op_resolver.h"
@@ -89,7 +90,8 @@ class TFLiteGPURunner {
serialized_binary_cache_ = std::move(cache);
}
std::vector<uint8_t> GetSerializedBinaryCache() {
absl::StatusOr<std::vector<uint8_t>> GetSerializedBinaryCache() {
RET_CHECK(cl_environment_) << "CL environment is not initialized.";
return cl_environment_->GetSerializedBinaryCache();
}