Skip to content

Commit

Permalink
gpu: ocl: fix padding issue for channel blocking in reduction
Browse files Browse the repository at this point in the history
  • Loading branch information
h-sadia authored and karturov committed May 19, 2022
1 parent f8de0c9 commit efbf9b5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/gpu/ocl/gen9_reduction.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -248,7 +248,7 @@ status_t gen9_reduction_t::pd_t::init_conf(engine_t *engine) {
} else {
conf.skip_final_phase = false;
initial_n_chunks_padded = conf.initial_n_chunks;
initial_c_padded = conf.src_dims[1];
initial_c_padded = utils::rnd_up(conf.src_dims[1], conf.c_block_size);
}

conf.dispatch.define_dim("INITIAL_N", 0, initial_n_chunks_padded, 1);
Expand Down

0 comments on commit efbf9b5

Please sign in to comment.