Skip to content

Commit 9010626

Browse files
committed
src: cpu: aarch64: Enable JIT fp16 -> fp16 reorder
1 parent bd119d5 commit 9010626

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/cpu/aarch64/jit_uni_reorder.cpp

+2-1
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,8 @@ struct jit_uni_reorder_kernel_f32_t : public kernel_t, public jit_generator {
172172

173173
bool is_f16 = (p.itype == f16 || p.otype == f16);
174174
bool f16_ok = (p.itype == f32 && p.otype == f16 && p.beta == 0.f)
175-
|| (p.itype == f16 && p.otype == f32 && p.beta == 0.f);
175+
|| (p.itype == f16 && p.otype == f32 && p.beta == 0.f)
176+
|| (p.itype == f16 && p.otype == f16 && p.beta == 0.f);
176177

177178
bool ok = true && p.ndims > 0
178179
&& utils::one_of(

0 commit comments

Comments
 (0)