p35189024
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix

This commit is contained in:
wanghailu 2024-01-02 05:52:04 +00:00
parent e3a2e65c47
commit 689f8c6a5d
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/kernels/bang/dropout.cc
View File

@ -41,6 +41,7 @@ class DropoutCnnl : public BangKernelWithoutConfig {
return; return;
checkCnnlError(cnnlDestroyTensorDescriptor(oDesc)); checkCnnlError(cnnlDestroyTensorDescriptor(oDesc));
checkCnnlError(cnnlDestroyTensorDescriptor(mDesc));
checkCnnlError(cnnlRandDestroyGenerator(generator)); checkCnnlError(cnnlRandDestroyGenerator(generator));
} }
}; };