• 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
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...
INSTALL Loading commit data...
LICENSE Loading commit data...
README-RESNET.rst Loading commit data...
README.md Loading commit data...
changes.md Loading commit data...