@@ -93,40 +93,40 @@ void typed_zero_pad_blk(const memory_desc_wrapper &m_d, void *data_handle) {
9393 if (c_tail_s) {
9494 parallel_nd (A, B, D, E, F,
9595 [=](dim_t a, dim_t b, dim_t d, dim_t e, dim_t f) {
96- auto x = &data[m_d.blk_off (a, b, C - 1 , d, e, f)];
97- if (blk_kind == c)
98- zeroize_tail (x, c_tail_s);
99- else if (blk_kind == bc)
100- zeroize_tail_inner (x, c_tail_s);
101- else if (blk_kind == cb)
102- zeroize_tail_outer (x, c_tail_s);
103- });
96+ auto x = &data[m_d.blk_off (a, b, C - 1 , d, e, f)];
97+ if (blk_kind == c)
98+ zeroize_tail (x, c_tail_s);
99+ else if (blk_kind == bc)
100+ zeroize_tail_inner (x, c_tail_s);
101+ else if (blk_kind == cb)
102+ zeroize_tail_outer (x, c_tail_s);
103+ });
104104 }
105105
106106 if (b_tail_s) {
107107 parallel_nd (A, C, D, E, F,
108108 [=](dim_t a, dim_t c, dim_t d, dim_t e, dim_t f) {
109- auto x = &data[m_d.blk_off (a, B - 1 , c, d, e, f)];
110- if (blk_kind == b)
111- zeroize_tail (x, b_tail_s);
112- else if (blk_kind == ab || blk_kind == cb)
113- zeroize_tail_inner (x, b_tail_s);
114- else if (blk_kind == ba || blk_kind == bc)
115- zeroize_tail_outer (x, b_tail_s);
116- });
109+ auto x = &data[m_d.blk_off (a, B - 1 , c, d, e, f)];
110+ if (blk_kind == b)
111+ zeroize_tail (x, b_tail_s);
112+ else if (blk_kind == ab || blk_kind == cb)
113+ zeroize_tail_inner (x, b_tail_s);
114+ else if (blk_kind == ba || blk_kind == bc)
115+ zeroize_tail_outer (x, b_tail_s);
116+ });
117117 }
118118
119119 if (a_tail_s) {
120120 parallel_nd (B, C, D, E, F,
121121 [=](dim_t b, dim_t c, dim_t d, dim_t e, dim_t f) {
122- auto x = &data[m_d.blk_off (A - 1 , b, c, d, e, f)];
123- if (blk_kind == a)
124- zeroize_tail (x, a_tail_s);
125- else if (blk_kind == ba)
126- zeroize_tail_inner (x, a_tail_s);
127- else if (blk_kind == ab)
128- zeroize_tail_outer (x, a_tail_s);
129- });
122+ auto x = &data[m_d.blk_off (A - 1 , b, c, d, e, f)];
123+ if (blk_kind == a)
124+ zeroize_tail (x, a_tail_s);
125+ else if (blk_kind == ba)
126+ zeroize_tail_inner (x, a_tail_s);
127+ else if (blk_kind == ab)
128+ zeroize_tail_outer (x, a_tail_s);
129+ });
130130 }
131131}
132132
0 commit comments