[CPU][Bugfix] Fix ppc64le CPU build (#30871)
Signed-off-by: Nishidha Panpaliya <nishidha.panpaliya@partner.ibm.com>
This commit is contained in:
committed by
GitHub
parent
420ba2dbb6
commit
bd2b52fc2d
@@ -37,10 +37,12 @@ struct VecTypeTrait<c10::BFloat16> {
|
|||||||
};
|
};
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
#if !defined(__powerpc__)
|
||||||
template <>
|
template <>
|
||||||
struct VecTypeTrait<c10::Half> {
|
struct VecTypeTrait<c10::Half> {
|
||||||
using vec_t = vec_op::FP16Vec16;
|
using vec_t = vec_op::FP16Vec16;
|
||||||
};
|
};
|
||||||
|
#endif
|
||||||
|
|
||||||
struct Counter {
|
struct Counter {
|
||||||
std::atomic<int64_t> counter;
|
std::atomic<int64_t> counter;
|
||||||
|
|||||||
Reference in New Issue
Block a user