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

Tensor: Cpu -> Ahwb storage transfer

Browse Source 
PiperOrigin-RevId: 494033280
This commit is contained in:
Nikolay Chirkov
2022-12-08 17:06:56 -08:00
committed by Copybara-Service
parent 05535db5f7
commit bea0caae65
4 changed files with 32 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/framework/formats/tensor_ahwb.cc
+1 -2
View File
@@ -214,7 +214,7 @@ Tensor::AHardwareBufferView Tensor::GetAHardwareBufferReadView() const {
"supported.";
CHECK(ahwb_ || !(valid_ & kValidOpenGlBuffer))
<< "Interoperability bettween OpenGL buffer and AHardwareBuffer is not "
"supported on targe system.";
"supported on target system.";
bool transfer = !ahwb_;
CHECK(AllocateAHardwareBuffer())
<< "AHardwareBuffer is not supported on the target system.";
@@ -268,7 +268,6 @@ Tensor::AHardwareBufferView Tensor::GetAHardwareBufferWriteView(
}
bool Tensor::AllocateAHardwareBuffer(int size_alignment) const {
if (!use_ahwb_) return false;
if (__builtin_available(android 26, *)) {
if (ahwb_ == nullptr) {
AHardwareBuffer_Desc desc = {};