@@ -527,11 +527,12 @@ where
527527macro_rules! all_reduce_func_def {
528528 ( $doc_str: expr, $fn_name: ident, $ffi_name: ident, $assoc_type: ident) => {
529529 #[ doc=$doc_str]
530- pub fn $fn_name<T >( input: & Array <T >)
531- -> (
532- <<T as HasAfEnum >:: $assoc_type as HasAfEnum >:: BaseType ,
533- <<T as HasAfEnum >:: $assoc_type as HasAfEnum >:: BaseType
534- )
530+ pub fn $fn_name<T >(
531+ input: & Array <T >,
532+ ) -> (
533+ <<T as HasAfEnum >:: $assoc_type as HasAfEnum >:: BaseType ,
534+ <<T as HasAfEnum >:: $assoc_type as HasAfEnum >:: BaseType ,
535+ )
535536 where
536537 T : HasAfEnum ,
537538 <T as HasAfEnum >:: $assoc_type: HasAfEnum ,
@@ -541,7 +542,9 @@ macro_rules! all_reduce_func_def {
541542 let mut imag: f64 = 0.0 ;
542543 unsafe {
543544 let err_val = $ffi_name(
544- & mut real as * mut c_double, & mut imag as * mut c_double, input. get( ) ,
545+ & mut real as * mut c_double,
546+ & mut imag as * mut c_double,
547+ input. get( ) ,
545548 ) ;
546549 HANDLE_ERROR ( AfError :: from( err_val) ) ;
547550 }
@@ -676,13 +679,15 @@ macro_rules! all_reduce_func_def2 {
676679 pub fn $fn_name<T >( input: & Array <T >) -> ( $out_type, $out_type)
677680 where
678681 T : HasAfEnum ,
679- $out_type: HasAfEnum + Fromf64
682+ $out_type: HasAfEnum + Fromf64 ,
680683 {
681684 let mut real: f64 = 0.0 ;
682685 let mut imag: f64 = 0.0 ;
683686 unsafe {
684687 let err_val = $ffi_name(
685- & mut real as * mut c_double, & mut imag as * mut c_double, input. get( ) ,
688+ & mut real as * mut c_double,
689+ & mut imag as * mut c_double,
690+ input. get( ) ,
686691 ) ;
687692 HANDLE_ERROR ( AfError :: from( err_val) ) ;
688693 }
@@ -869,13 +874,16 @@ macro_rules! dim_ireduce_func_def {
869874 T :: $out_type: HasAfEnum ,
870875 {
871876 unsafe {
872- let mut temp: af_array = std:: ptr:: null_mut( ) ;
873- let mut idx: af_array = std:: ptr:: null_mut( ) ;
877+ let mut temp: af_array = std:: ptr:: null_mut( ) ;
878+ let mut idx: af_array = std:: ptr:: null_mut( ) ;
874879 let err_val = $ffi_name(
875- & mut temp as * mut af_array, & mut idx as * mut af_array, input. get( ) , dim,
880+ & mut temp as * mut af_array,
881+ & mut idx as * mut af_array,
882+ input. get( ) ,
883+ dim,
876884 ) ;
877885 HANDLE_ERROR ( AfError :: from( err_val) ) ;
878- ( temp. into( ) , idx. into( ) )
886+ ( temp. into( ) , idx. into( ) )
879887 }
880888 }
881889 } ;
@@ -910,12 +918,13 @@ dim_ireduce_func_def!("
910918macro_rules! all_ireduce_func_def {
911919 ( $doc_str: expr, $fn_name: ident, $ffi_name: ident, $assoc_type: ident) => {
912920 #[ doc=$doc_str]
913- pub fn $fn_name<T >( input: & Array <T >)
914- -> (
915- <<T as HasAfEnum >:: $assoc_type as HasAfEnum >:: BaseType ,
916- <<T as HasAfEnum >:: $assoc_type as HasAfEnum >:: BaseType ,
917- u32
918- )
921+ pub fn $fn_name<T >(
922+ input: & Array <T >,
923+ ) -> (
924+ <<T as HasAfEnum >:: $assoc_type as HasAfEnum >:: BaseType ,
925+ <<T as HasAfEnum >:: $assoc_type as HasAfEnum >:: BaseType ,
926+ u32 ,
927+ )
919928 where
920929 T : HasAfEnum ,
921930 <T as HasAfEnum >:: $assoc_type: HasAfEnum ,
@@ -926,8 +935,10 @@ macro_rules! all_ireduce_func_def {
926935 let mut temp: u32 = 0 ;
927936 unsafe {
928937 let err_val = $ffi_name(
929- & mut real as * mut c_double, & mut imag as * mut c_double,
930- & mut temp as * mut c_uint, input. get( ) ,
938+ & mut real as * mut c_double,
939+ & mut imag as * mut c_double,
940+ & mut temp as * mut c_uint,
941+ input. get( ) ,
931942 ) ;
932943 HANDLE_ERROR ( AfError :: from( err_val) ) ;
933944 }
@@ -1277,23 +1288,28 @@ macro_rules! dim_reduce_by_key_func_def {
12771288 /// Tuple of Arrays, with output keys and values after reduction
12781289 ///
12791290 #[ doc=$ex_str]
1280- pub fn $fn_name<KeyType , ValueType >( keys: & Array <KeyType >, vals: & Array <ValueType >,
1281- dim: i32
1291+ pub fn $fn_name<KeyType , ValueType >(
1292+ keys: & Array <KeyType >,
1293+ vals: & Array <ValueType >,
1294+ dim: i32 ,
12821295 ) -> ( Array <KeyType >, Array <$out_type>)
12831296 where
12841297 KeyType : ReduceByKeyInput ,
12851298 ValueType : HasAfEnum ,
12861299 $out_type: HasAfEnum ,
12871300 {
12881301 unsafe {
1289- let mut out_keys: af_array = std:: ptr:: null_mut( ) ;
1290- let mut out_vals: af_array = std:: ptr:: null_mut( ) ;
1302+ let mut out_keys: af_array = std:: ptr:: null_mut( ) ;
1303+ let mut out_vals: af_array = std:: ptr:: null_mut( ) ;
12911304 let err_val = $ffi_name(
1292- & mut out_keys as * mut af_array, & mut out_vals as * mut af_array,
1293- keys. get( ) , vals. get( ) , dim,
1305+ & mut out_keys as * mut af_array,
1306+ & mut out_vals as * mut af_array,
1307+ keys. get( ) ,
1308+ vals. get( ) ,
1309+ dim,
12941310 ) ;
12951311 HANDLE_ERROR ( AfError :: from( err_val) ) ;
1296- ( out_keys. into( ) , out_vals. into( ) )
1312+ ( out_keys. into( ) , out_vals. into( ) )
12971313 }
12981314 }
12991315 } ;
@@ -1408,24 +1424,30 @@ macro_rules! dim_reduce_by_key_nan_func_def {
14081424 /// Tuple of Arrays, with output keys and values after reduction
14091425 ///
14101426 #[ doc=$ex_str]
1411- pub fn $fn_name<KeyType , ValueType >( keys: & Array <KeyType >, vals: & Array <ValueType >,
1412- dim: i32 , replace_value: f64
1427+ pub fn $fn_name<KeyType , ValueType >(
1428+ keys: & Array <KeyType >,
1429+ vals: & Array <ValueType >,
1430+ dim: i32 ,
1431+ replace_value: f64 ,
14131432 ) -> ( Array <KeyType >, Array <$out_type>)
14141433 where
14151434 KeyType : ReduceByKeyInput ,
14161435 ValueType : HasAfEnum ,
14171436 $out_type: HasAfEnum ,
14181437 {
14191438 unsafe {
1420- let mut out_keys: af_array = std:: ptr:: null_mut( ) ;
1421- let mut out_vals: af_array = std:: ptr:: null_mut( ) ;
1439+ let mut out_keys: af_array = std:: ptr:: null_mut( ) ;
1440+ let mut out_vals: af_array = std:: ptr:: null_mut( ) ;
14221441 let err_val = $ffi_name(
14231442 & mut out_keys as * mut af_array,
14241443 & mut out_vals as * mut af_array,
1425- keys. get( ) , vals. get( ) , dim, replace_value,
1444+ keys. get( ) ,
1445+ vals. get( ) ,
1446+ dim,
1447+ replace_value,
14261448 ) ;
14271449 HANDLE_ERROR ( AfError :: from( err_val) ) ;
1428- ( out_keys. into( ) , out_vals. into( ) )
1450+ ( out_keys. into( ) , out_vals. into( ) )
14291451 }
14301452 }
14311453 } ;
0 commit comments