From 689f8c6a5d00e17dc477fce84122a4afc3242313 Mon Sep 17 00:00:00 2001 From: wanghailu Date: Tue, 2 Jan 2024 05:52:04 +0000 Subject: [PATCH] fix --- src/kernels/bang/dropout.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/src/kernels/bang/dropout.cc b/src/kernels/bang/dropout.cc index b8b3eafb..5b52f5ce 100644 --- a/src/kernels/bang/dropout.cc +++ b/src/kernels/bang/dropout.cc @@ -41,6 +41,7 @@ class DropoutCnnl : public BangKernelWithoutConfig { return; checkCnnlError(cnnlDestroyTensorDescriptor(oDesc)); + checkCnnlError(cnnlDestroyTensorDescriptor(mDesc)); checkCnnlError(cnnlRandDestroyGenerator(generator)); } };