Skip to content

Commit

Permalink
feat: Don't print the tensor in inspect
Browse files Browse the repository at this point in the history
  • Loading branch information
uttarayan21 committed Sep 5, 2024
1 parent 104d34e commit 2fd2f42
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions examples/inspect.rs
Original file line number Diff line number Diff line change
Expand Up @@ -61,8 +61,7 @@ pub fn main() -> anyhow::Result<()> {
let tensor = x.tensor::<f32>().expect("No tensor");
time!(tensor.wait(ffi::MapType::MAP_TENSOR_READ, true); format!("Waiting for tensor: {}", x.name()));
println!("{}: {:?}", x.name(), tensor.shape());
let cpu_tensor = tensor.create_host_tensor_from_device(true);
println!("{:?}", cpu_tensor.host());
let _ = tensor.create_host_tensor_from_device(true);
// std::fs::write(format!("{}.bin", x.name()), bytemuck::cast_slice(cpu_tensor.host())).expect("Unable to write");
});
// Ok(())
Expand Down

0 comments on commit 2fd2f42

Please sign in to comment.