Fix GL Tensor (backed by AHardwaBuffer) GL to CPU synchronization.
glFinish() must be called on a valid gl context PiperOrigin-RevId: 478816905
This commit is contained in:
parent
8d5cf9bbed
commit
38baaa00b1
|
@ -371,7 +371,7 @@ void* Tensor::MapAhwbToCpuRead() const {
|
||||||
if ((valid_ & kValidOpenGlBuffer) && ssbo_written_ == -1) {
|
if ((valid_ & kValidOpenGlBuffer) && ssbo_written_ == -1) {
|
||||||
// EGLSync is failed. Use another synchronization method.
|
// EGLSync is failed. Use another synchronization method.
|
||||||
// TODO: Use tflite::gpu::GlBufferSync and GlActiveSync.
|
// TODO: Use tflite::gpu::GlBufferSync and GlActiveSync.
|
||||||
glFinish();
|
gl_context_->Run([]() { glFinish(); });
|
||||||
} else if (valid_ & kValidAHardwareBuffer) {
|
} else if (valid_ & kValidAHardwareBuffer) {
|
||||||
CHECK(ahwb_written_) << "Ahwb-to-Cpu synchronization requires the "
|
CHECK(ahwb_written_) << "Ahwb-to-Cpu synchronization requires the "
|
||||||
"completion function to be set";
|
"completion function to be set";
|
||||||
|
|
Loading…
Reference in New Issue
Block a user