• P
    [cherry-pick][Paddle-TRT] Stack op plugin (#25605) (#27365) · 4283be52
    Pei Yang 提交于
    * [Paddle-TRT] Stack op plugin (#25605)
    
    * add stack_op to CMakeLists
    
    * add dim=3 support for scale op
    
    * add trt stack op, test=develop
    
    * remove debug message
    
    * add stack plugin serialize
    
    * remove slice, scale op, will add later
    
    * enhence error message
    
    * revise trt ernie test to conver the stack op CI testi, test=develop
    
    * add stack op serialization
    
    * fix test shape after adding stack op
    
    * remove slice op, will add after implementing serialization
    
    * roll back to min_graph=5 to avoid using slice op
    
    * fix scale op output layer
    
    * implement stack op createPlugin
    
    * use workspace and move the defination to .cu
    
    * move stack plugin creator definition to .cu, test=develop
    
    * sync ut with develop
    Co-authored-by: Nzlsh80826 <zlsh80826@gmail.com>
    4283be52
op_teller.cc 3.8 KB