From 71ca4459d9e79d13c05e6ea0359da6d68d8c50a2 Mon Sep 17 00:00:00 2001 From: YdrMaster Date: Fri, 24 Feb 2023 15:04:31 +0800 Subject: [PATCH] fmt Signed-off-by: YdrMaster --- src/kernels/cuda/batch_norm.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/kernels/cuda/batch_norm.cc b/src/kernels/cuda/batch_norm.cc index b150aaa5..35cc78c0 100644 --- a/src/kernels/cuda/batch_norm.cc +++ b/src/kernels/cuda/batch_norm.cc @@ -2,6 +2,7 @@ #include "core/kernel.h" #include "cuda/cuda_kernel_wihtout_config.h" #include "cuda/cuda_runtime.h" + namespace infini { class BatchNormCudnn : public CudaKernelWithoutConfig { void compute(const Operator &_op, @@ -29,7 +30,7 @@ class BatchNormCudnn : public CudaKernelWithoutConfig { dimArray[i] = dims[i]; strideArray[i] = op->getInputs(0)->getStride()[i]; dimPArray[i] = 1; - stridePArray[i] = 1; + stridePArray[i] = 1; } dimPArray[1] = op->getInputs(0)->getDims()[1]; stridePArray[1] = op->getInputs(0)->getStride()[1];