Update source/api_cc/src/DeepPotJAX.cc

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Nov 4, 2024 · f16dd92 · f16dd92
1 parent 2aa6deb
commit f16dd92
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/source/api_cc/src/DeepPotJAX.cc b/source/api_cc/src/DeepPotJAX.cc
@@ -123,6 +123,7 @@ inline std::vector<T> get_vector(TFE_Context* ctx,
   std::vector<T> result;
   tensor_to_vector(result, retval, status);
   // deallocate
+  TFE_DeleteTensorHandle(retval);
   TFE_DeleteOp(op);
   return result;
 }