From c8f6290c03df9b433637c4c78457132abdaf6f26 Mon Sep 17 00:00:00 2001 From: wanghailu Date: Fri, 30 Jun 2023 14:56:18 +0800 Subject: [PATCH] fix code --- src/kernels/bang/clip.cc | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/kernels/bang/clip.cc b/src/kernels/bang/clip.cc index 4f1cb640..a93cb963 100644 --- a/src/kernels/bang/clip.cc +++ b/src/kernels/bang/clip.cc @@ -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;