Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Potential memory leak #791

Open
prasannagn16 opened this issue Oct 8, 2024 · 0 comments
Open

Potential memory leak #791

prasannagn16 opened this issue Oct 8, 2024 · 0 comments

Comments

@prasannagn16
Copy link

Description
I am leveraging Triton Inference Server r24.07.
While trying to analyse the code , I see some potential memory leak for the some of the below file ,

1.common.cc

           ` *infer_input = new InferInput(name, dims, datatype); `

2.common.cc

        `*infer_output = new InferRequestedOutput(name, datatype, class_count);` 

3.torchserve_infer_input.cc

	`TorchServeInferInput* local_infer_input = new TorchServeInferInput(name, dims, datatype); `

4.torchserve_http_client.cc

	`*infer_result = reinterpret_cast<InferResult*>(new InferResult(infer_request));`

5.openai_infer_input.cc

	 `OpenAiInferInput* local_infer_input = new OpenAiInferInput(name, dims, datatype); `

6.grpc_client.cc

	 `*infer_result = reinterpret_cast<InferResult*>(new InferResultGrpc(response, request_status));` 

7.grpc_client.cc

	  `*infer_result = reinterpret_cast<InferResult*>(new InferResultGrpc(response));	`

I think here not handle the memory deallocation, it’s directly passing the raw pointer .

Consider one example :
common.cc

Error
InferInput::Create(
    InferInput** infer_input, const std::string& name,
    const std::vector<int64_t>& dims, const std::string& datatype)
{
  *infer_input = new InferInput(name, dims, datatype);
  return Error::Success;
}

InferInput** infer_input is pointer to a pointer (i.e., a raw pointer). It holds the address of a pointer to InferInput

Here its create the “create” method as a static .

common.h

class InferInput {
 public:
  /// Create a InferInput instance that describes a model input.
  /// \param infer_input Returns a new InferInput object.
  /// \param name The name of input whose data will be described by this object.
  /// \param dims The shape of the input.
  /// \param datatype The datatype of the input.
  /// \return Error object indicating success or failure.
  static Error Create(
      InferInput** infer_input, const std::string& name,
      const std::vector<int64_t>& dims, const std::string& datatype);
};

client_backend.h

class InferInput {
 public:
  /// Create a InferInput instance that describes a model input.
  /// \param infer_input Returns a new InferInput object.
  /// \param kind The kind of the associated client backend.
  /// \param name The name of input whose data will be described by this object.
  /// \param dims The shape of the input.
  /// \param datatype The datatype of the input.
  /// \return Error object indicating success or failure.
  static Error Create(
      InferInput** infer_input, const BackendKind kind, const std::string& name,
      const std::vector<int64_t>& dims, const std::string& datatype);

  virtual ~InferInput() = default;
};

For the above code I have understand that ,
this class designed in factory design pattern function but still since passing raw pointer ,if caller's or user as forgot handle the memory deallocation it leads to memory leak.

Solution is:

  • Use while creating “create“ method parameter for smart pointer or handle the memory deallocation in the raw pointer .

  • Handel the raw pointer deallocation using delete.

Triton Information r24.07
The code designed such way that as per the requirement user need to handle the memory .Is it documented for any user guide?

Expected behaviour

  • Handle the memory deallocation for raw pointer using delete or else use smart pointer.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Development

No branches or pull requests

1 participant