• S
    [PHI] Migrate softmax kernel (#47339) · b8ae3858
    Sławomir Siwek 提交于
    * add extra attr property set
    
    * add type_info for all context
    
    * add onednn context to all context
    
    * fix context compile error
    
    * simplify conv kernel args
    
    * pass runtime attr into dev_ctx
    
    * fix marco error
    
    * clear conv_grad_kernel extra args
    
    * merge conv_grad_grad into conv_grad
    
    * clear conv2d_grad_grad extra attrs
    
    * remove redundant imports
    
    * migrate softmax
    
    * clear yaml and eager extra attr
    
    * fix conv1d error
    
    * change to thread local
    
    * fix npu compile failed
    
    * try to fix windows compile failed
    
    * add conv2d onednn phi kernel
    
    * fix ci bugs (#36)
    
    * fix compile bugs (#38)
    
    * fix extra input transform bug (#39)
    
    * support dynamic created attr (#40)
    
    * reset extra info gen code
    
    * rm conv_grad_grad kernel
    
    * reimpl pass attr adapting
    
    * add int attr support
    
    * remove vector inputnames creating
    
    * merge dev
    
    * fix map at error
    
    * adjust attribute
    
    * adapt funcs to PHI
    Co-authored-by: NChen Weihang <chenweihang@baidu.com>
    Co-authored-by: NYuanRisheng <yuanrisheng@baidu.com>
    b8ae3858
test_mkldnn_caching.cc 5.8 KB