Skip to content

Commit 304c93f

Browse files
committed
xe: conv: limit tanh workaround to ATS-M/DG2 only
1 parent 343c291 commit 304c93f

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/gpu/intel/jit/conv/config.cpp

+5-2
Original file line numberDiff line numberDiff line change
@@ -885,8 +885,11 @@ bool post_ops_ok(const conv_problem_t &prb, const hw_t &hw) {
885885
if (!jit_eltwise_injector_f32_is_supported(po.eltwise.alg))
886886
return false;
887887
else if (po.eltwise.alg == alg_kind::eltwise_tanh
888-
&& hw == ngen::HW::XeHPG && hw.systolic_support()
889-
&& hw.eu_count() <= 128)
888+
&& hw == ngen::HW::XeHPG
889+
&& utils::one_of(hw.product_family(),
890+
ngen::ProductFamily::GenericXeHPG,
891+
ngen::ProductFamily::DG2)
892+
&& hw.systolic_support() && hw.eu_count() <= 128)
890893
// Workaround for hard to reproduce issue in end to end
891894
// workloads. It is unclear what the actual issue is as the
892895
// kernel always works correctly in benchdnn.

0 commit comments

Comments
 (0)