chore: remove arc

huggingface · Jan 22, 2024 · 5691f47 · 5691f47
1 parent a762bf5
commit 5691f47
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/crates/ratchet-core/src/tensor.rs b/crates/ratchet-core/src/tensor.rs
@@ -468,6 +468,15 @@ def matmul(a, b):
         });
         println!("Successfully created ground truth");
         println!("{:?}", ground);
+
+        let device = Device::request_device(DeviceRequest::GPU)?;
+        let a_gpu = a.to(device.clone())?;
+        let b_gpu = b.to(device.clone())?;
+        let c_gpu = a_gpu.matmul(&b_gpu)?;
+        c_gpu.resolve()?;
+        let d_gpu = c_gpu.to(Device::CPU)?;
+        println!("Successfully resolved GPU tensor");
+        println!("{:?}", d_gpu);
         Ok(())
     }
 }