Added invalidate handle marco

Type: Code Improvement Signed-off-by: Feiyue Chen <[email protected]>
VeriSilicon · Dec 1, 2022 · 7c331a9 · 7c331a9
1 parent c231c54
commit 7c331a9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/tim/vx/tensor.cc b/src/tim/vx/tensor.cc
@@ -172,6 +172,9 @@ bool TensorImpl::CopyDataFromTensor(void* data) {
       if (tensor->attr.is_created_from_handle) {
         void* ptr = NULL;
         vsi_nn_GetTensorHandle(tensor, &ptr);
+        #ifdef VSI_INVALIDATE_HANDLE_SUPPORT
+            vsi_nn_InvalidateHandle(tensor);
+        #endif
         if (ptr) {
           memcpy(data, ptr, tensor_bytes);
           retn = true;