chore: dbg

huggingface · Jan 22, 2024 · 3a761bc · 3a761bc
1 parent cb08cbd
commit 3a761bc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/crates/ratchet-core/src/tensor.rs b/crates/ratchet-core/src/tensor.rs
@@ -416,6 +416,7 @@ mod tests {
 
     use super::*;
 
+    /*
     #[test]
     fn test_matmul() -> anyhow::Result<()> {
         let device = Device::request_device(DeviceRequest::GPU)?;
@@ -426,6 +427,7 @@ mod tests {
         let d = c.to(Device::CPU)?;
         Ok(())
     }
+    */
 
     #[test]
     fn test_pyo3() -> anyhow::Result<()> {
@@ -448,6 +450,7 @@ mod tests {
                 "x",
             )?;
 
+            println!("About to send tensors to python");
             let py_a = a.to_py::<f32>(&py);
             println!("Converted A to pyo3 array");
             let py_b = b.to_py::<f32>(&py);