• J
    【Prim】Custom softmax grad (#51474) · f124c86f
    Jiabin Yang 提交于
    * [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
    
    * Pr 50885 (#7)
    
    * [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>
    
    * [prim] enable dygraph_to_static to support custom_vjp
    
    * fix cast prim and vjp dtype mapping error bug
    
    * Cxx prim custom vjp (#8)
    
    * [CINN]Enhance CacheKey hash logic by considering input dtypes (#50557)
    
    ---------
    Co-authored-by: Njiangcheng <thisjiang@qq.com>
    
    * [prim] enable dygraph_to_static to support custom_vjp
    
    * Pr 50885 (#7)
    
    * [CINN]Enhance CacheKey hash logic by considering input dtypes (#50557)
    
    * [CINN]Enhance CacheKey hash logic by considering input dtypes
    
    ---------
    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>
    
    * [prim] enable dygraph_to_static to support custom_vjp
    
    * fix cast prim and vjp dtype mapping error bug
    
    * [dy2static-ci] fix dy2static ci errors.
    
    ---------
    Co-authored-by: NAurelius84 <zhangliujie@baidu.com>
    Co-authored-by: Njiangcheng <thisjiang@qq.com>
    Co-authored-by: Ncxxly <chenxx_id@163.com>
    
    * [Prim] enable whitelist and blacklist for custom_vjp
    
    * support softmax grad
    
    * remove additional code
    
    * add test back
    
    ---------
    Co-authored-by: NAurelius84 <zhangliujie@baidu.com>
    Co-authored-by: Njiangcheng <thisjiang@qq.com>
    Co-authored-by: Ncxxly <chenxx_id@163.com>
    Co-authored-by: Nxiongkun <807377414@qq.com>
    f124c86f
legacy_backward.yaml 56.2 KB