• 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
..
ngraph Loading commit data...
resource Loading commit data...
CMakeLists.txt Loading commit data...