Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "96b9aa5aa076e64c68765232aec343e4d0006e2a"
Unverified Commit 4de87866 authored by R3hankhan's avatar R3hankhan Committed by GitHub
Browse files

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


Signed-off-by: default avatarRehan Khan <Rehan.Khan7@ibm.com>
parent eca7a8fb
Expand all Show whitespace changes
Inline Side-by-side
Showing with 531 additions and 57 deletions
csrc/cpu/cpu_attn_impl.hpp
View file @ 4de87866
...@@ -847,7 +847,7 @@ struct VecTypeTrait<c10::BFloat16> { ...@@ -847,7 +847,7 @@ struct VecTypeTrait<c10::BFloat16> {
}; };
#endif #endif
#if !defined(__powerpc__) #if !defined(__powerpc__) && !defined(__s390x__)
template <> template <>
struct VecTypeTrait<c10::Half> { struct VecTypeTrait<c10::Half> {
using vec_t = vec_op::FP16Vec16; using vec_t = vec_op::FP16Vec16;
......
csrc/cpu/cpu_types_vxe.hpp
View file @ 4de87866
This diff is collapsed.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment