• Tristan Webb's avatar
    GPU kernels for reshape, GEMM, EW ADD/Mult, Maximum (#440) · da50410b
    Tristan Webb authored
    * GPU kernels for reshape, GEMM, EW ADD/Mult, Maximum
    
    (A + B) * C test now with cuBLAS
    Additional gemm and gemv calls
    cmake updates for cuDNN calls
    memcpy wrappers in gpu_util
    
    Additional passing tests:
    aliased outputs, parameter, constant tensor memcopy
    da50410b
cudnn.cpp 7.49 KB