• S
    Move fused batchnorm to Phi (#53476) · 5e5481d8
    Sonder 提交于
    * trans fused batch norm Compute function
    
    * trans batch norm register info to phi
    
    * trans fused batch norm grad Compute
    
    * trans batch norm grad register info
    
    * add sig file
    
    * update sig file
    
    * Update fused_bn_activation_kernel.cu
    
    * Update fused_bn_activation_grad_kernel.cu
    
    * fix
    
    * Rename fused_bn_activation_kernel_grad.cu to fused_bn_activation_kernel.cu
    
    * fix
    
    * fix
    
    * fix CudnnDataType error
    
    * fix
    
    * fix include
    
    * update
    
    * add #if
    
    * add fused bn act to cmakelist.txt
    
    * update  cmakelist
    
    * fix #ifdef error
    
    * add timeout set
    
    * add env set
    
    * fix
    
    * fix
    
    * Update fused_bn_activation_sig.cc
    5e5481d8
fused_bn_activation_kernel.h 1.6 KB