File tree 1 file changed +18
-7
lines changed
1 file changed +18
-7
lines changed Original file line number Diff line number Diff line change @@ -890,13 +890,24 @@ where
890
890
T : HasAfEnum ,
891
891
{
892
892
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
+ }
900
911
}
901
912
}
902
913
You can’t perform that action at this time.
0 commit comments