Skip to content

Commit 6431cba

Browse files
committed
Improve Array Debug trait output message
1 parent 1fdbc75 commit 6431cba

File tree

1 file changed

+18
-7
lines changed

1 file changed

+18
-7
lines changed

src/core/array.rs

Lines changed: 18 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -890,13 +890,24 @@ where
890890
T: HasAfEnum,
891891
{
892892
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
893-
let mut vec = vec![T::default(); self.elements()];
894-
self.host(&mut vec);
895-
f.debug_struct("Array")
896-
.field("dtype", &self.get_type())
897-
.field("shape", &self.dims())
898-
.field("data", &vec)
899-
.finish()
893+
if f.alternate() {
894+
let mut vec = vec![T::default(); self.elements()];
895+
self.host(&mut vec);
896+
f.debug_struct("Array")
897+
.field("dtype", &self.get_type())
898+
.field("shape", &self.dims())
899+
.field("strides", &self.strides())
900+
.field("offset", &self.offset())
901+
.field("device_id", &self.get_device_id())
902+
.field("data", &vec)
903+
.finish()
904+
} else {
905+
f.debug_struct("Array")
906+
.field("dtype", &self.get_type())
907+
.field("shape", &self.dims())
908+
.field("af_array", unsafe { &self.get() })
909+
.finish()
910+
}
900911
}
901912
}
902913

0 commit comments

Comments
 (0)