We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 343c291 commit 304c93fCopy full SHA for 304c93f
src/gpu/intel/jit/conv/config.cpp
@@ -885,8 +885,11 @@ bool post_ops_ok(const conv_problem_t &prb, const hw_t &hw) {
885
if (!jit_eltwise_injector_f32_is_supported(po.eltwise.alg))
886
return false;
887
else if (po.eltwise.alg == alg_kind::eltwise_tanh
888
- && hw == ngen::HW::XeHPG && hw.systolic_support()
889
- && hw.eu_count() <= 128)
+ && hw == ngen::HW::XeHPG
+ && utils::one_of(hw.product_family(),
890
+ ngen::ProductFamily::GenericXeHPG,
891
+ ngen::ProductFamily::DG2)
892
+ && hw.systolic_support() && hw.eu_count() <= 128)
893
// Workaround for hard to reproduce issue in end to end
894
// workloads. It is unclear what the actual issue is as the
895
// kernel always works correctly in benchdnn.
0 commit comments