File tree 2 files changed +12
-8
lines changed
2 files changed +12
-8
lines changed Original file line number Diff line number Diff line change @@ -52,10 +52,12 @@ struct jit_brgemm_amx_uker_base_t : public jit_generator {
52
52
for (int i = 0 ; i < post_ops.len (); i++) {
53
53
const auto &entry = post_ops.entry_ [i];
54
54
if (!entry.is_binary ()) continue ;
55
- has_f8_e5m2_binary_postops = entry.binary .src1_desc .data_type
56
- == data_type::f8_e5m2;
57
- has_f8_e4m3_binary_postops = entry.binary .src1_desc .data_type
58
- == data_type::f8_e4m3;
55
+ has_f8_e5m2_binary_postops
56
+ = entry.binary .src1_desc .data_type == data_type::f8_e5m2
57
+ || has_f8_e5m2_binary_postops;
58
+ has_f8_e4m3_binary_postops
59
+ = entry.binary .src1_desc .data_type == data_type::f8_e4m3
60
+ || has_f8_e4m3_binary_postops;
59
61
}
60
62
}
61
63
Original file line number Diff line number Diff line change @@ -64,10 +64,12 @@ struct jit_brgemm_kernel_t : public jit_generator {
64
64
for (int i = 0 ; i < post_ops.len (); i++) {
65
65
const auto &entry = post_ops.entry_ [i];
66
66
if (!entry.is_binary ()) continue ;
67
- has_f8_e5m2_binary_postops = entry.binary .src1_desc .data_type
68
- == data_type::f8_e5m2;
69
- has_f8_e4m3_binary_postops = entry.binary .src1_desc .data_type
70
- == data_type::f8_e4m3;
67
+ has_f8_e5m2_binary_postops
68
+ = entry.binary .src1_desc .data_type == data_type::f8_e5m2
69
+ || has_f8_e5m2_binary_postops;
70
+ has_f8_e4m3_binary_postops
71
+ = entry.binary .src1_desc .data_type == data_type::f8_e4m3
72
+ || has_f8_e4m3_binary_postops;
71
73
}
72
74
}
73
75
You can’t perform that action at this time.
0 commit comments