• rocking's avatar
    MaxPool & AvgPool bwd instances, test, ckProfiler, client example (#861) · 866377de
    rocking authored
    * Add maxpool instances
    
    * Rename index pool to max pool.
    
    * Add maxpool bwd bf16 instances
    
    * Add avg pool bwd instances
    
    * Rename avgpool and maxpool to avg_pool3d and max_pool
    
    * Add bf16 pool fwd instances
    
    * Add max pool bwd to ckProfiler
    
    * Add avg pool3d bwd to ckProfiler
    
    * Add avg pool bwd test
    
    * Fix bug of reference pool fwd (dilation)
    
    * Fix bug of max pool bwd  (dilation and initZero)
    
    * Support bf16 compute data type
    
    * Force compute type be f32. Because atomicAdd only support f32
    
    * Add max pool bwd test
    
    * Rename folder
    
    * Rename pool
    
    * Add max pool bwd client example
    
    * Add avg pool bwd client example
    
    * Add missing workspace
    
    * clang format
    
    * Rename macro
    
    * remove useless header
    
    * remove useless layout
    866377de
test_max_pool3d_fwd.cpp 4.37 KB