Skip to content

Commit 1135306

Browse files
fix: fix compute cap matching (#21)
1 parent db4ff2b commit 1135306

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

backends/candle/src/compute_cap.rs

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,11 @@ pub fn incompatible_compute_cap() -> bool {
2222
let compile_compute_cap = *COMPILE_COMPUTE_CAP;
2323
let runtime_compute_cap = *RUNTIME_COMPUTE_CAP;
2424

25-
((runtime_compute_cap == 75 || runtime_compute_cap == 90)
26-
&& runtime_compute_cap != compile_compute_cap)
27-
|| (!((80..90).contains(&runtime_compute_cap) && (80..90).contains(&compile_compute_cap)))
25+
match (runtime_compute_cap, compile_compute_cap) {
26+
(75, 75) => false,
27+
(80, 80) => false,
28+
(86, 80..=86) => false,
29+
(90, 90) => false,
30+
(_, _) => true,
31+
}
2832
}

router/src/main.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -231,7 +231,7 @@ async fn main() -> Result<()> {
231231
);
232232

233233
// Get dtype
234-
let dtype = args.dtype.unwrap_or_else(|| {
234+
let dtype = args.dtype.unwrap_or({
235235
#[cfg(feature = "accelerate")]
236236
{
237237
DType::Float32

0 commit comments

Comments
 (0)