Skip to content

Commit b547b0b

Browse files
committed
Fix feature-gating for DType::Bfloat16 (and exclude Turing and Volta)
1 parent b1a1b23 commit b547b0b

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

router/src/lib.rs

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -269,8 +269,14 @@ pub async fn run(
269269
.unwrap_or_default()
270270
});
271271

272+
#[cfg(any(
273+
feature = "python",
274+
any(all(feature = "candle", feature = "metal"), feature = "candle-cuda")
275+
))]
272276
if dtype == DType::Bfloat16 {
273-
tracing::warn!("`--dtype bfloat16` support is still experimental on Metal and CUDA.");
277+
tracing::warn!(
278+
"`--dtype bfloat16` support is still experimental on Metal and CUDA (Ampere onwards)."
279+
);
274280
}
275281

276282
tracing::info!("Starting model backend");

0 commit comments

Comments
 (0)