• Tristan Webb's avatar
    GPU kernels for reshape, GEMM, EW ADD/Mult, Maximum · 1f6284ff
    Tristan Webb authored
    GPU ew add and mult cuBLAS calls
    
    GPU (A + B) * C with cuBLAS
    
    Additional gemm and gemv calls
    
    cmake updates for cuDNN calls
    
    kernels WIP
    
    params for dot gemm
    
    more kernel WIP
    
    memcpy wrappers
    
    aliased outputs, parameter, constant tensor memcopy
    
    comment cleanup
    
    remove cruft
    
    gpu faster gemm
    
    MNIST WIP
    
    Cleanup
    1f6284ff
Name
Last commit
Last update
cmake Loading commit data...
contrib/docker Loading commit data...
doc Loading commit data...
maint Loading commit data...
src Loading commit data...
test Loading commit data...
third-party Loading commit data...
.clang-format Loading commit data...
.gitignore Loading commit data...
CMakeLists.txt Loading commit data...
README-RESNET.rst Loading commit data...
README.md Loading commit data...
changes.md Loading commit data...