Skip to content

Commit 43a8e99

Browse files
committed
Add note on lack of BF16 support for Turing (and earlier)
1 parent b547b0b commit 43a8e99

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

backends/src/dtype.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,8 @@ pub enum DType {
1515
Float32,
1616
#[cfg(any(
1717
feature = "python",
18+
// NOTE: It needs to exclude Turing (and Volta) i.e., anything before Ampere, as it won't
19+
// have support for BF16
1820
all(feature = "candle", any(feature = "metal", feature = "cuda"))
1921
))]
2022
Bfloat16,

0 commit comments

Comments
 (0)