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

fix: 修改 clip kernel 

Signed-off-by: YdrMaster <ydrml@hotmail.com>
This commit is contained in:
YdrMaster 2023-06-30 14:26:56 +08:00
parent d6338feb1b
commit 1a925ba5c7
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,6 +1,7 @@
#include "bang/bang_kernel_without_config.h"
#include "bang/bang_runtime.h"
#include "operators/unary.h"
#include <cmath>
namespace infini {
class ClipCnnl : public BangKernelWithoutConfig {
@ -22,7 +23,9 @@ class ClipCnnl : public BangKernelWithoutConfig {
CNNL_DTYPE_FLOAT, aDim.size(),
aDim.data()));
cnnlStatus_t stat =
cnnlClip(context->cnnlHandle(), aDesc, aData, &min, &max, cData);
cnnlClip(context->cnnlHandle(), aDesc, aData,
std::isfinite(min) ? &min : nullptr,
std::isfinite(max) ? &max : nullptr, cData);
if (stat != CNNL_STATUS_SUCCESS)
return;