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

Signed-off-by: Rehan Khan <[email protected]>

Author: R3hankhan123

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;