• C
    [Pten] Refactor the implementation of custom operator (#37122) · 1e598f1a
    Chen Weihang 提交于
    * move extension into pten [no-verify]
    
    * append tensor methods by ext_tensor [no-verify]
    
    * append other tensor methods [no-verify]
    
    * ext related files tidy [no-verify]
    
    * include relation tidy [no-verify]
    
    * add pten tensor test [no-verify]
    
    * replace tensor in custom op & compile success
    
    * refine tensor constructor for unittest
    
    * custom relu jit run success
    
    * fix all custom op unittests
    
    * add inference cmake adapt [no-verify]
    
    * fix failed unittests
    
    * fix windows failed unittests
    
    * try to fix kunlun and inference failed
    
    * fix test_elementwise_api error
    
    * try to fix win compile failed
    
    * fix kunlun fp16 type error
    
    * remove useless haddle error macro
    
    * add custom linear op test
    
    * fix compile failed & add win symbols
    
    * fix non pten kernel cast failed
    
    * add dll decl for api
    
    * polish several deetails
    
    * polish details by review comment
    
    * add dll_decl for register
    1e598f1a
custom_relu_op.cu 2.7 KB