File tree Expand file tree Collapse file tree 1 file changed +18
-7
lines changed Expand file tree Collapse file tree 1 file changed +18
-7
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments