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

Signed-off-by: Rehan Khan <Rehan.Khan7@ibm.com>

[CPU][IBM Z] Fix BF16 support and vectorize math operations for s390x (#28926)
Signed-off-by: Rehan Khan <Rehan.Khan7@ibm.com>
4de87866 · R3hankhan · GitHub · eca7a8fb · 4de87866 · 4de87866
Unverified Commit 4de87866 authored Nov 24, 2025 by R3hankhan Committed by GitHub Nov 24, 2025
Expand all Hide whitespace changes
Inline Side-by-side

Showing with 531 additions and 57 deletions

csrc/cpu/cpu_attn_impl.hpp csrc/cpu/cpu_attn_impl.hpp +1 -1

csrc/cpu/cpu_types_vxe.hpp csrc/cpu/cpu_types_vxe.hpp +530 -56

No files found.
--- 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;

--- a/csrc/cpu/cpu_types_vxe.hpp
+++ b/csrc/cpu/cpu_types_vxe.hpp