• Chao Liu's avatar
    Tweak GEMM kernel (#38) · b3e8d57d
    Chao Liu authored
    * add parameters
    
    * tweak gemm
    
    * tweak
    
    * update conv
    
    * update script
    
    * adding bwd 1x1
    
    * update script
    
    * adding 1x1 bwd
    
    * debugging bwd 1x1 failure
    
    * update script
    
    * update script
    
    * test
    
    * test v100
    
    * clean up
    b3e8d57d
config.hpp 3.96 KB