• A
    [PluggableDevice] Add custom kernel support based on pten kernel management (#38848) · a8879215
    Aganlengzi 提交于
    * [Demo] custom kernel based on pten kernel
    
    * merge and npu custom work well
    
    * del comments
    
    * delete other code
    
    * fix CUDAContext
    
    * fix not found small_vector.h
    
    * support NPU
    
    * fix NPUContext
    
    * fix DeviceContext support
    
    * add UT
    
    * fix call
    
    * add UT
    
    * fix
    
    * fix for comments and ut
    
    * add MACRO control
    
    * fix multi input output
    
    * support env CUSTOM_DEVICE_ROOT
    
    * deal with special cases
    
    * fix for Windows
    
    * try coverage with test_custom_kernel_dot.py
    
    * fix test_custom_kernel_dot
    
    * fix test_custom_kernel_dot
    
    * fix merge
    
    * fix merge
    
    * fix CI
    
    * update
    
    * merge and fix
    
    * remove WITH_CUSTOM_KERNEL
    
    * fix merge
    
    * merge and fix
    
    * fix ut
    
    * fix ut for mac
    
    * add more UT
    
    * add more UT
    
    * fix
    a8879215
custom_kernel_test.cc 11.5 KB