• Y
    [Semi-Auto] Add reshape spmd rule (#55177) · a97b507e
    Yichen Zhang 提交于
    * add reshape spmd rule
    
    * add unit test for reshape spmd rule
    
    * bug fix
    
    * replace the print_info function with to_string
    
    * fix typo
    
    * bug fix
    
    * add handling for "0" in target shape
    
    * remove the part of computing size in dim_trans.cc
    a97b507e
dim_trans.cc 11.4 KB