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 {
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}
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments