Skip to content

Commit f070d5c

Browse files
committed
Improve Array Debug trait output message
1 parent 33d0996 commit f070d5c

File tree

1 file changed

+15
-7
lines changed

1 file changed

+15
-7
lines changed

src/core/array.rs

Lines changed: 15 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -890,13 +890,21 @@ 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("data", &vec)
900+
.finish()
901+
} else {
902+
f.debug_struct("Array")
903+
.field("dtype", &self.get_type())
904+
.field("shape", &self.dims())
905+
.field("af_array", unsafe { &self.get() })
906+
.finish()
907+
}
900908
}
901909
}
902910

0 commit comments

Comments
 (0)