• rocking's avatar
    Maxpool bwd (#750) · 341ad956
    rocking authored
    * Add maxpool f32 kernel and example
    
    * Revise copyright
    
    * Add device pool bwd device op
    
    * Support f16 and bf16
    
    * Add compute datatype for reference code.
    Prevent error in bf16
    
    * Fix type error
    
    * Remove layout
    
    * Fix bf16 error
    
    * Add f16 and bf16 example
    
    * Add more operations
    
    * Implement IsSupportedArgument
    
    * Add changelog
    
    * Add comment
    
    * Add comment
    
    * Remove useless header
    
    * Move initialize of workspace to the run
    
    * Move set din zero to the device operator
    
    * Save din_length_raw
    
    * Remove useless header
    
    * Calculate gridsize according to the number of CU
    
    * Calculate gridSize according to the number of CU.
    Remove useless header
    
    * Add put example
    
    * Remove useless header
    
    * Fix CI fail
    341ad956
maxpool2d_bwd_bf16.cpp 2.16 KB