• C
    [pten] add split kernel (#39060) · d0df5632
    chentianyu03 提交于
    * add split kernel
    
    * add split kernel signature
    
    * fix split bug
    
    * modify MakePtenScalarArrayFromVarList
    
    * modify MakePtenScalarArrayFromVarList
    
    * fix split windows register error
    
    * add test case for split kernel
    
    * replace raw split kernel with pten kernel
    
    * fix makeScalar/ScalarArray bug
    
    * remove debug log
    
    * remove int64_t type in buildPtcontext
    
    * update by code review
    
    * fix split dev test failed
    
    * change DenseTensorMeta to MetaTensor
    
    * change split api code from auto gen to manual
    
    * split cuda kernel support bfloat16 type
    
    * fix conflict
    
    * rm raw split kernel
    
    * merge develop branch
    
    * change to pten::errors
    d0df5632
test_split_api.cc 2.8 KB