[CPU][IBM Z] Fix BF16 support and vectorize math operations for s390x (#28926)

Signed-off-by: Rehan Khan <Rehan.Khan7@ibm.com>
2025-11-24 17:38:09 +05:30
parent eca7a8fb59
commit 4de87866a8
2 changed files with 531 additions and 57 deletions
--- a/csrc/cpu/cpu_attn_impl.hpp
+++ b/csrc/cpu/cpu_attn_impl.hpp
@@ -847,7 +847,7 @@ struct VecTypeTrait<c10::BFloat16> {
 };
 #endif

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