Skip to content

Commit 76c2783

Browse files
committed
gpu: bnorm: nhwc-reusable: rename some kernels
1 parent 26cea44 commit 76c2783

File tree

4 files changed

+8
-8
lines changed

4 files changed

+8
-8
lines changed

src/gpu/ocl/bnorm/bnorm_utils.hpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -74,14 +74,14 @@ constexpr int aux_fwd = 1;
7474
constexpr int aux_bwd = 0;
7575

7676
namespace kernel_id {
77-
constexpr size_t update_fwd = 0;
77+
constexpr size_t norm_fwd = 0;
7878
constexpr size_t calc_mean = 1;
7979
constexpr size_t calc_var = 2;
8080
constexpr size_t reduce_fwd_reg = 3;
8181
constexpr size_t calc_mean_var = 4;
8282
constexpr size_t reduce_fwd_1pass = 5;
8383
constexpr size_t reduce_aux = 6;
84-
constexpr size_t update_bwd = 7;
84+
constexpr size_t norm_bwd = 7;
8585
constexpr size_t calc_stat = 8;
8686
constexpr size_t reduce_stat = 9;
8787
} // namespace kernel_id

src/gpu/ocl/bnorm/nhwc_reusable.cl

+2-2
Original file line numberDiff line numberDiff line change
@@ -336,7 +336,7 @@ nhwc_reusable_calc_mean_var(__global DATA_T *src, __global float *reduce_temp,
336336

337337
// Main FWD kernel, common for regular and 1pass algorithms
338338
__attribute__((intel_reqd_sub_group_size(16))) __kernel void
339-
nhwc_reusable_update_fwd(__global DATA_T *src, __global float *mean,
339+
nhwc_reusable_norm_fwd(__global DATA_T *src, __global float *mean,
340340
__global float *variance, __global DATA_T *dst,
341341
__global float *scaleshift, __global float *shift, __global char *ws,
342342
float eps, __global DATA_T *src_add, float relu_alpha, off_t ic_size,
@@ -632,7 +632,7 @@ nhwc_reusable_calc_stat(__global DATA_T *src, __global float *mean,
632632

633633
// Main BWD pass kernel
634634
__attribute__((intel_reqd_sub_group_size(16))) __kernel void
635-
nhwc_reusable_update_bwd(__global DATA_T *src, __global float *mean,
635+
nhwc_reusable_norm_bwd(__global DATA_T *src, __global float *mean,
636636
__global float *variance, __global DATA_T *diff_dst,
637637
__global float *scaleshift, __global char *ws,
638638
__global DATA_T *diff_src, __global float *diff_scale,

src/gpu/ocl/bnorm/nhwc_reusable.cpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -495,7 +495,7 @@ status_t nhwc_reusable_batch_normalization_fwd_t::execute_forward(
495495
arg_list.append(rt_conf.update_sp_block);
496496

497497
auto nd_range = pd()->dispatch.nd_range();
498-
return parallel_for(ctx, nd_range, kernels_[update_fwd], arg_list);
498+
return parallel_for(ctx, nd_range, kernels_[norm_fwd], arg_list);
499499
}
500500

501501
status_t nhwc_reusable_batch_normalization_bwd_t::pd_t::init_conf(
@@ -639,7 +639,7 @@ status_t nhwc_reusable_batch_normalization_bwd_t::execute_backward(
639639
arg_list.append(rt_conf.update_sp_block);
640640

641641
auto nd_range = pd()->dispatch.nd_range();
642-
return parallel_for(ctx, nd_range, kernels_[update_bwd], arg_list);
642+
return parallel_for(ctx, nd_range, kernels_[norm_bwd], arg_list);
643643
}
644644

645645
} // namespace ocl

src/gpu/ocl/bnorm/nhwc_reusable.hpp

+2-2
Original file line numberDiff line numberDiff line change
@@ -50,10 +50,10 @@ struct nhwc_reusable_bnorm_compile_params_t {
5050

5151
const std::vector<const char *> &get_kernel_names() const {
5252
static const std::vector<const char *> kernel_names = {
53-
"nhwc_reusable_update_fwd", "nhwc_reusable_calc_mean",
53+
"nhwc_reusable_norm_fwd", "nhwc_reusable_calc_mean",
5454
"nhwc_reusable_calc_var", "nhwc_reusable_reduce_fwd_reg",
5555
"nhwc_reusable_calc_mean_var", "nhwc_reusable_reduce_fwd_1pass",
56-
"nhwc_reusable_reduce_aux", "nhwc_reusable_update_bwd",
56+
"nhwc_reusable_reduce_aux", "nhwc_reusable_norm_bwd",
5757
"nhwc_reusable_calc_stat", "nhwc_reusable_reduce_stat"};
5858
return kernel_names;
5959
}

0 commit comments

Comments
 (0)