diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-07-08 08:39:27 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-08 08:39:27 +0100 |
| commit | 33479c5f1b98a6e9f537ea139449bb8dc26fed3e (patch) | |
| tree | d4b78d20f7ce5457a8596d46a92f39ad184b672a /candle-core/src/display.rs | |
| parent | f35cfc5e976d454541dcc66934aa97969a08595f (diff) | |
| download | candle-33479c5f1b98a6e9f537ea139449bb8dc26fed3e.tar.gz candle-33479c5f1b98a6e9f537ea139449bb8dc26fed3e.tar.bz2 candle-33479c5f1b98a6e9f537ea139449bb8dc26fed3e.zip | |
Add some very simple sum benchmark. (#108)
* Add some very simple sum benchmark.
* Rename the file.
Diffstat (limited to 'candle-core/src/display.rs')
| -rw-r--r-- | candle-core/src/display.rs | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/candle-core/src/display.rs b/candle-core/src/display.rs index 60907bb3..127e55b0 100644 --- a/candle-core/src/display.rs +++ b/candle-core/src/display.rs @@ -9,7 +9,11 @@ impl Tensor { &self, f: &mut std::fmt::Formatter, ) -> std::fmt::Result { - write!(f, "Tensor[")?; + let prefix = match self.device() { + crate::Device::Cpu => "Cpu", + crate::Device::Cuda(_) => "Cuda", + }; + write!(f, "{prefix}Tensor[")?; match self.dims() { [] => { if let Ok(v) = self.to_scalar::<T>() { |