1. 16 5月, 2023 1 次提交
    • 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