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

fix code

This commit is contained in:
wanghailu 2023-06-30 14:56:18 +08:00
parent 1a925ba5c7
commit c8f6290c03
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/kernels/bang/clip.cc
View File

@ -22,10 +22,16 @@ class ClipCnnl : public BangKernelWithoutConfig {
checkCnnlError(cnnlSetTensorDescriptor(aDesc, CNNL_LAYOUT_NCHW,
CNNL_DTYPE_FLOAT, aDim.size(),
aDim.data()));
cnnlStatus_t stat =
cnnlStatus_t stat;
if ( std::isfinite(min) || std::isfinite(max)) {
stat =
cnnlClip(context->cnnlHandle(), aDesc, aData,
std::isfinite(min) ? &min : nullptr,
std::isfinite(max) ? &max : nullptr, cData);
} else {
stat =
cnnlCopy(context->cnnlHandle(), aDesc, aData, aDesc, cData);
}
if (stat != CNNL_STATUS_SUCCESS)
return;