clang-format formatting of the ml_dtypes directory.

jax-ml · Sep 4, 2024 · c864dc6 · c864dc6
1 parent ea94484
commit c864dc6
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 15 deletions.
diff --git a/ml_dtypes/_src/dtypes.cc b/ml_dtypes/_src/dtypes.cc
@@ -69,7 +69,7 @@ struct TypeDescriptor<float8_e3m4> : CustomFloatType<float8_e3m4> {
   static constexpr const char* kQualifiedTypeName = "ml_dtypes.float8_e3m4";
   static constexpr const char* kTpDoc = "float8_e3m4 floating-point values";
   // Set e3m4 kind as Void since kind=f (float) with itemsize=1 is used by e5m2
-  static constexpr char kNpyDescrKind = 'V';       // Void
+  static constexpr char kNpyDescrKind = 'V';  // Void
   static constexpr char kNpyDescrType = '3';
   static constexpr char kNpyDescrByteorder = '=';  // Native byte order
 };

diff --git a/ml_dtypes/include/float8.h b/ml_dtypes/include/float8.h
@@ -255,8 +255,7 @@ class float8_e3m4 : public float8_base<float8_e3m4> {
 
  public:
   template <typename T, RequiresIsDerivedFromFloat8Base<T> = 0>
-  explicit EIGEN_DEVICE_FUNC float8_e3m4(T f8)
-      : float8_e3m4(ConvertFrom(f8)) {}
+  explicit EIGEN_DEVICE_FUNC float8_e3m4(T f8) : float8_e3m4(ConvertFrom(f8)) {}
 };
 
 class float8_e4m3 : public float8_base<float8_e4m3> {
@@ -269,8 +268,7 @@ class float8_e4m3 : public float8_base<float8_e4m3> {
 
  public:
   template <typename T, RequiresIsDerivedFromFloat8Base<T> = 0>
-  explicit EIGEN_DEVICE_FUNC float8_e4m3(T f8)
-      : float8_e4m3(ConvertFrom(f8)) {}
+  explicit EIGEN_DEVICE_FUNC float8_e4m3(T f8) : float8_e4m3(ConvertFrom(f8)) {}
 };
 
 class float8_e4m3fn : public float8_base<float8_e4m3fn> {
@@ -1481,25 +1479,25 @@ namespace numext {
 
 template <>
 EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC ml_dtypes::float8_e3m4
-bit_cast<ml_dtypes::float8_e3m4, uint8_t>(const uint8_t &src) {
+bit_cast<ml_dtypes::float8_e3m4, uint8_t>(const uint8_t& src) {
   return ml_dtypes::float8_e3m4::FromRep(src);
 }
 
 template <>
 EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC uint8_t
-bit_cast<uint8_t, ml_dtypes::float8_e3m4>(const ml_dtypes::float8_e3m4 &src) {
+bit_cast<uint8_t, ml_dtypes::float8_e3m4>(const ml_dtypes::float8_e3m4& src) {
   return src.rep();
 }
 
 template <>
 EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC ml_dtypes::float8_e4m3
-bit_cast<ml_dtypes::float8_e4m3, uint8_t>(const uint8_t &src) {
+bit_cast<ml_dtypes::float8_e4m3, uint8_t>(const uint8_t& src) {
   return ml_dtypes::float8_e4m3::FromRep(src);
 }
 
 template <>
 EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC uint8_t
-bit_cast<uint8_t, ml_dtypes::float8_e4m3>(const ml_dtypes::float8_e4m3 &src) {
+bit_cast<uint8_t, ml_dtypes::float8_e4m3>(const ml_dtypes::float8_e4m3& src) {
   return src.rep();
 }
 

diff --git a/ml_dtypes/tests/float8_test.cc b/ml_dtypes/tests/float8_test.cc
@@ -65,8 +65,7 @@ TEST(Float8E3m4Test, NumericLimits) {
       Eigen::numext::isnan(std::numeric_limits<float8_e3m4>::quiet_NaN()));
   EXPECT_TRUE(
       Eigen::numext::isnan(std::numeric_limits<float8_e3m4>::signaling_NaN()));
-  EXPECT_EQ(static_cast<float>(std::numeric_limits<float8_e3m4>::min()),
-            0.25);
+  EXPECT_EQ(static_cast<float>(std::numeric_limits<float8_e3m4>::min()), 0.25);
   EXPECT_EQ(static_cast<float>(std::numeric_limits<float8_e3m4>::max()), 15.5);
   EXPECT_EQ(static_cast<float>(std::numeric_limits<float8_e3m4>::lowest()),
             -15.5);
@@ -995,10 +994,10 @@ struct Float8CastTestParamNames {
       std::pair<Type, float8_e5m2>, std::pair<Type, bool>,                 \
       std::pair<Type, int32_t>, std::pair<Type, int64_t>
 
-#define GEN_TYPE_PAIRS()                                                       \
-  GEN_DEST_TYPES(float8_e3m4), GEN_DEST_TYPES(float8_e4m3),                    \
-      GEN_DEST_TYPES(float8_e4m3fn), GEN_DEST_TYPES(float8_e4m3b11fnuz),       \
-      GEN_DEST_TYPES(float8_e5m2), GEN_DEST_TYPES(float8_e4m3fnuz),            \
+#define GEN_TYPE_PAIRS()                                                 \
+  GEN_DEST_TYPES(float8_e3m4), GEN_DEST_TYPES(float8_e4m3),              \
+      GEN_DEST_TYPES(float8_e4m3fn), GEN_DEST_TYPES(float8_e4m3b11fnuz), \
+      GEN_DEST_TYPES(float8_e5m2), GEN_DEST_TYPES(float8_e4m3fnuz),      \
       GEN_DEST_TYPES(float8_e5m2fnuz)
 
 using Float8CastTypePairs = ::testing::Types<GEN_TYPE_PAIRS()>;