mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
44cc873fba
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56830 Opt into formatting on GitHub and format everything. This is a trial run before turning on formatting for more and eventually all of the codebase. Test Plan: CI Reviewed By: zertosh Differential Revision: D27979080 fbshipit-source-id: a80f0c48691c08ae8ca0af06377b87e6a2351151
29 lines
866 B
C++
29 lines
866 B
C++
#pragma once
|
|
|
|
#include <c10/util/BFloat16.h>
|
|
#include <c10/util/Half.h>
|
|
#include <c10/util/math_compat.h>
|
|
|
|
namespace c10 {
|
|
|
|
// Note: Explicit implementation of copysign for Half and BFloat16
|
|
// is needed to workaround g++-7/8 crash on aarch64, but also makes
|
|
// copysign faster for the half-precision types
|
|
template <typename T, typename U>
|
|
inline auto copysign(const T& a, const U& b) {
|
|
return std::copysign(a, b);
|
|
}
|
|
|
|
// Implement copysign for half precision floats using bit ops
|
|
// Sign is the most significant bit for both half and bfloat16 types
|
|
inline c10::Half copysign(c10::Half a, c10::Half b) {
|
|
return c10::Half((a.x & 0x7fff) | (b.x & 0x8000), c10::Half::from_bits());
|
|
}
|
|
|
|
inline c10::BFloat16 copysign(c10::BFloat16 a, c10::BFloat16 b) {
|
|
return c10::BFloat16(
|
|
(a.x & 0x7fff) | (b.x & 0x8000), c10::BFloat16::from_bits());
|
|
}
|
|
|
|
} // namespace c10
|