Skip to content

Commit 4dee389

Browse files
nshustrovakharito
authored andcommitted
cpu: avx512_core: disbale AutoGrow mode in fp32 winograd convolutions
1 parent 7904759 commit 4dee389

File tree

2 files changed

+4
-7
lines changed

2 files changed

+4
-7
lines changed

src/cpu/jit_avx512_common_conv_winograd_kernel_f32.hpp

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,7 @@ constexpr int simd_w = 16;
3636
struct _jit_avx512_common_conv_winograd_data_kernel_f32 : public jit_generator {
3737
_jit_avx512_common_conv_winograd_data_kernel_f32(
3838
jit_conv_winograd_conf_t ajcp)
39-
: jcp(ajcp)
40-
{
39+
: jit_generator(nullptr, MAX_CODE_SIZE, false), jcp(ajcp) {
4140
//******************* First iter kernel ********************//
4241
this->gemm_loop_generate(true);
4342
gemm_loop_ker_first_iter
@@ -111,8 +110,7 @@ struct jit_avx512_common_conv_winograd_bwd_weights_kernel_f32
111110

112111
jit_avx512_common_conv_winograd_bwd_weights_kernel_f32(
113112
jit_conv_winograd_conf_t ajcp)
114-
: jcp(ajcp)
115-
{
113+
: jit_generator(nullptr, MAX_CODE_SIZE, false), jcp(ajcp) {
116114

117115
//******************* First iter kernel ********************//
118116
{

src/cpu/jit_avx512_core_fp32_wino_conv_4x3_kernel.hpp

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ struct _jit_avx512_core_fp32_wino_conv_4x3_data_kernel
3333
: public jit_generator {
3434
_jit_avx512_core_fp32_wino_conv_4x3_data_kernel(
3535
jit_conv_winograd_conf_t ajcp)
36-
: jcp(ajcp) {
36+
: jit_generator(nullptr, MAX_CODE_SIZE, false), jcp(ajcp) {
3737
{
3838
this->weights_transform_data_ker_generate();
3939
weights_transform_data_ker
@@ -182,8 +182,7 @@ struct jit_avx512_core_fp32_wino_conv_4x3_bwd_weights_kernel
182182

183183
jit_avx512_core_fp32_wino_conv_4x3_bwd_weights_kernel(
184184
jit_conv_winograd_conf_t ajcp)
185-
: jcp(ajcp)
186-
{
185+
: jit_generator(nullptr, MAX_CODE_SIZE, false), jcp(ajcp) {
187186
//******************* First iter kernel ********************//
188187
this->gemm_loop_generate(true);
189188
gemm_loop_ker_first_iter = (decltype(gemm_loop_ker_first_iter))this->getCode();

0 commit comments

Comments
 (0)