add half::16 support

Author: swfsql

Cargo.toml

@@ -44,6 +44,8 @@ matrixmultiply = { version = "0.3.2", default-features = false, features=["cgemm
 serde = { version = "1.0", optional = true, default-features = false, features = ["alloc"] }
 rawpointer = { version = "0.2" }
 
+half = {version = "2.7.1", default-features = false, features = ["num-traits"], optional = true}
+
 [dev-dependencies]
 defmac = "0.2"
 quickcheck = { workspace = true }
@@ -67,6 +69,8 @@ matrixmultiply-threading = ["matrixmultiply/threading"]
 
 portable-atomic-critical-section = ["portable-atomic/critical-section"]
 
+half = ["dep:half"]
+
 
 [target.'cfg(not(target_has_atomic = "ptr"))'.dependencies]
 portable-atomic = { version = "1.6.0" }

src/impl_ops.rs

@@ -45,8 +45,16 @@ impl ScalarOperand for i128 {}
 impl ScalarOperand for u128 {}
 impl ScalarOperand for isize {}
 impl ScalarOperand for usize {}
+#[cfg(feature = "half")]
+impl ScalarOperand for half::f16 {}
+#[cfg(feature = "half")]
+impl ScalarOperand for half::bf16 {}
 impl ScalarOperand for f32 {}
 impl ScalarOperand for f64 {}
+#[cfg(feature = "half")]
+impl ScalarOperand for Complex<half::f16> {}
+#[cfg(feature = "half")]
+impl ScalarOperand for Complex<half::bf16> {}
 impl ScalarOperand for Complex<f32> {}
 impl ScalarOperand for Complex<f64> {}
 
@@ -468,6 +476,26 @@ mod arithmetic_ops
     impl_scalar_lhs_op!(bool, Commute, |, BitOr, bitor, "bit or");
     impl_scalar_lhs_op!(bool, Commute, ^, BitXor, bitxor, "bit xor");
 
+    #[cfg(feature = "half")]
+    mod ops_f16 {
+        use super::*;
+        impl_scalar_lhs_op!(half::f16, Commute, +, Add, add, "addition");
+        impl_scalar_lhs_op!(half::f16, Ordered, -, Sub, sub, "subtraction");
+        impl_scalar_lhs_op!(half::f16, Commute, *, Mul, mul, "multiplication");
+        impl_scalar_lhs_op!(half::f16, Ordered, /, Div, div, "division");
+        impl_scalar_lhs_op!(half::f16, Ordered, %, Rem, rem, "remainder");
+    }
+
+    #[cfg(feature = "half")]
+    mod ops_bf16 {
+        use super::*;
+        impl_scalar_lhs_op!(half::bf16, Commute, +, Add, add, "addition");
+        impl_scalar_lhs_op!(half::bf16, Ordered, -, Sub, sub, "subtraction");
+        impl_scalar_lhs_op!(half::bf16, Commute, *, Mul, mul, "multiplication");
+        impl_scalar_lhs_op!(half::bf16, Ordered, /, Div, div, "division");
+        impl_scalar_lhs_op!(half::bf16, Ordered, %, Rem, rem, "remainder");
+    }
+
     impl_scalar_lhs_op!(f32, Commute, +, Add, add, "addition");
     impl_scalar_lhs_op!(f32, Ordered, -, Sub, sub, "subtraction");
     impl_scalar_lhs_op!(f32, Commute, *, Mul, mul, "multiplication");
@@ -480,6 +508,24 @@ mod arithmetic_ops
     impl_scalar_lhs_op!(f64, Ordered, /, Div, div, "division");
     impl_scalar_lhs_op!(f64, Ordered, %, Rem, rem, "remainder");
 
+    #[cfg(feature = "half")]
+    mod ops_complex_f16 {
+        use super::*;
+        impl_scalar_lhs_op!(Complex<half::f16>, Commute, +, Add, add, "addition");
+        impl_scalar_lhs_op!(Complex<half::f16>, Ordered, -, Sub, sub, "subtraction");
+        impl_scalar_lhs_op!(Complex<half::f16>, Commute, *, Mul, mul, "multiplication");
+        impl_scalar_lhs_op!(Complex<half::f16>, Ordered, /, Div, div, "division");
+    }
+
+    #[cfg(feature = "half")]
+    mod ops_complex_bf16 {
+        use super::*;
+        impl_scalar_lhs_op!(Complex<half::bf16>, Commute, +, Add, add, "addition");
+        impl_scalar_lhs_op!(Complex<half::bf16>, Ordered, -, Sub, sub, "subtraction");
+        impl_scalar_lhs_op!(Complex<half::bf16>, Commute, *, Mul, mul, "multiplication");
+        impl_scalar_lhs_op!(Complex<half::bf16>, Ordered, /, Div, div, "division");
+    }
+
     impl_scalar_lhs_op!(Complex<f32>, Commute, +, Add, add, "addition");
     impl_scalar_lhs_op!(Complex<f32>, Ordered, -, Sub, sub, "subtraction");
     impl_scalar_lhs_op!(Complex<f32>, Commute, *, Mul, mul, "multiplication");

src/linalg_traits.rs

@@ -61,6 +61,14 @@ pub trait NdFloat:
 {
 }
 
+#[cfg(all(feature = "std", feature = "half"))]
+#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
+impl NdFloat for half::f16 {}
+
+#[cfg(all(feature = "std", feature = "half"))]
+#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
+impl NdFloat for half::bf16 {}
+
 #[cfg(feature = "std")]
 #[cfg_attr(docsrs, doc(cfg(feature = "std")))]
 impl NdFloat for f32 {}