Skip to content

There might be memory leak in instanceNormalizationPlugin  #296

@NHZlX

Description

@NHZlX

There are two places that can cause leaks

First, every enqueue calls cudamalloc without releasing

CHECK_CUDA(cudaMalloc((void**) &_d_bias, n * nchan_bytes));

Second, cudnn_handle will also be created every time

CHECK_CUDNN(cudnnCreate(&_cudnn_handle));

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions