"megatron/vscode:/vscode.git/clone" did not exist on "be136206d2d58e2c18665d169602f9222662826b"
  • 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
docker-rocm4.3.1.sh 767 Bytes