• X
    Pr 50885 (#7) · ecc842f1
    xiongkun 提交于
    * [CINN]Enhance CacheKey hash logic by considering input dtypes (#50557)
    
    * [CINN]Enhance CacheKey hash logic by considering input dtypes
    
    * add unittest
    
    * fix typo
    
    * fix typo
    
    * fix map.at
    
    * fix find
    
    * fix test
    
    * fix cinn cache key structure realize
    
    * using ordered map for attributes
    
    * add test by review advice
    
    ---------
    Co-authored-by: Njiangcheng <thisjiang@qq.com>
    
    * [prim] enable dygraph_to_static to support custom_vjp
    
    * fix code in a dy2static-friendly way.
    
    * [dystatic] add hooker for prim
    
    ---------
    Co-authored-by: NAurelius84 <zhangliujie@baidu.com>
    Co-authored-by: Njiangcheng <thisjiang@qq.com>
    Co-authored-by: Ncxxly <chenxx_id@163.com>
    ecc842f1
partial_program.py 41.0 KB