disable f16, since we don't support it yet

coreylowman · Dec 3, 2023 · 21c9f62 · 21c9f62
1 parent 2c212e5
commit 21c9f62
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/dfdx-core/src/tensor_ops/utilities/device.rs b/dfdx-core/src/tensor_ops/utilities/device.rs
@@ -131,10 +131,11 @@ impl Device<f32> for crate::tensor::Cuda {}
 #[cfg(feature = "cuda")]
 impl Device<f64> for crate::tensor::Cuda {}
 
-#[cfg(all(feature = "webgpu", feature = "f16"))]
-impl Device<f16> for crate::tensor::Webgpu {}
-#[cfg(all(feature = "webgpu", feature = "f16"))]
-impl Device<AMP<f16>> for crate::tensor::Webgpu {}
+// TODO: How can we implement this for f16 when WGSL doesn't support f16 yet?
+// #[cfg(all(feature = "webgpu", feature = "f16"))]
+// impl Device<f16> for crate::tensor::Webgpu {}
+// #[cfg(all(feature = "webgpu", feature = "f16"))]
+// impl Device<AMP<f16>> for crate::tensor::Webgpu {}
 #[cfg(feature = "webgpu")]
 impl Device<f32> for crate::tensor::Webgpu {}