fix

2024-01-02 05:52:04 +00:00 · 2024-01-02 05:52:04 +00:00 · 689f8c6a5d
parent e3a2e65c47
commit 689f8c6a5d
1 changed files with 1 additions and 0 deletions
--- 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));
    }
 };