fix(ggml): only set backend after if guards

rustformers · Jul 16, 2023 · d815857 · d815857
1 parent 090735a
commit d815857
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/crates/ggml/src/tensor.rs b/crates/ggml/src/tensor.rs
@@ -65,10 +65,10 @@ impl Tensor {
             if current_backend != Backend::Cpu && backend == Backend::Cpu {
                 unimplemented!("Tensors cannot be moved from an accelerator to the CPU at present");
             }
-            t.set_backend(backend);
             if backend == Backend::Cpu {
                 return;
             }
+            t.set_backend(backend);
 
             #[cfg(feature = "cublas")]
             unsafe {