[CPU] Enable FP16 (Half dtype) support for s390x (#34116)

Signed-off-by: Rehan Khan <Rehan.Khan7@ibm.com>
2026-02-11 12:11:42 +05:30
parent 0b20469c62
commit d1b837f0ae
3 changed files with 244 additions and 9 deletions
--- a/csrc/cpu/cpu_attn_impl.hpp
+++ b/csrc/cpu/cpu_attn_impl.hpp
@@ -821,7 +821,7 @@ struct VecTypeTrait<c10::BFloat16> {
  using vec_t = vec_op::BF16Vec16;
 };

-#if !defined(__powerpc__) && !defined(__s390x__)
+#if !defined(__powerpc__)
 template <>
 struct VecTypeTrait<c10::Half> {
  using vec_t = vec_op::FP16Vec16;