change padding size back to 4

This commit is contained in:
bssrdf 2025-10-15 14:21:04 -04:00
parent ac77b8d0e0
commit 6a1f8b4d57
1 changed files with 1 additions and 1 deletions

View File

@ -25,7 +25,7 @@ static __global__ void reduce_f32(const float * __restrict__ x, float * __restri
template<typename T, const int BM, const int BN, const int BK, const int WM, const int WN,
const int WNITER, const int TM, const int TN, const int NUM_THREADS,
// layout: 0, NHWC; 1, NCHW
const int layout, const bool vec_load, const int ksplit, const int PAD=1>
const int layout, const bool vec_load, const int ksplit, const int PAD=4>
static __global__ void conv2d_implicit_kernel(const float * __restrict__ input,
const T * __restrict__ kernel,
float * __restrict__ output,