File tree Expand file tree Collapse file tree 2 files changed +8
-4
lines changed Expand file tree Collapse file tree 2 files changed +8
-4
lines changed Original file line number Diff line number Diff line change @@ -22,7 +22,11 @@ pub fn incompatible_compute_cap() -> bool {
22
22
let compile_compute_cap = * COMPILE_COMPUTE_CAP ;
23
23
let runtime_compute_cap = * RUNTIME_COMPUTE_CAP ;
24
24
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
+ }
28
32
}
Original file line number Diff line number Diff line change @@ -231,7 +231,7 @@ async fn main() -> Result<()> {
231
231
) ;
232
232
233
233
// Get dtype
234
- let dtype = args. dtype . unwrap_or_else ( || {
234
+ let dtype = args. dtype . unwrap_or ( {
235
235
#[ cfg( feature = "accelerate" ) ]
236
236
{
237
237
DType :: Float32
You can’t perform that action at this time.
0 commit comments