saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-08 00:23:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix mem leak (#10272)

Browse source
This commit is contained in:
Vincent Wang 2022-01-14 14:54:19 +08:00 committed by GitHub
parent 2a55bc2c21
commit 3ea7fb0f9f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
orttraining/orttraining/python/training/ortmodule/torch_cpp_extensions/cpu/aten_op_executor/aten_op_executor.cc
View file

@ -68,9 +68,11 @@ struct ATenOperator {
bool is_list = is_list_arguments[index];
c10::IValue i_value;
// Create the torch tensor from this DLPack no matter we need it or not below,
// so that the dlpack's deleter will be triggered when torch tensor is out of scope.
at::Tensor tensor = at::fromDLPack(dlpack);
switch (elem_kinds[index]) {
case c10::TypeKind::TensorType: {
at::Tensor tensor = at::fromDLPack(dlpack);
i_value = is_optional ? c10::IValue(c10::optional<at::Tensor>(tensor)) : c10::IValue(tensor);
} break;
case c10::TypeKind::IntType: {