• 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
..
autodiff Loading commit data...
builder Loading commit data...
codegen Loading commit data...
descriptor Loading commit data...
ops Loading commit data...
pass Loading commit data...
pattern Loading commit data...
runtime Loading commit data...
types Loading commit data...
CMakeLists.txt Loading commit data...
common.cpp Loading commit data...
common.hpp Loading commit data...
coordinate_transform.cpp Loading commit data...
coordinate_transform.hpp Loading commit data...
except.hpp Loading commit data...
file_util.cpp Loading commit data...
file_util.hpp Loading commit data...
function.cpp Loading commit data...
function.hpp Loading commit data...
graph_util.cpp Loading commit data...
graph_util.hpp Loading commit data...
json.hpp Loading commit data...
log.cpp Loading commit data...
log.hpp Loading commit data...
ngraph.hpp Loading commit data...
node.cpp Loading commit data...
node.hpp Loading commit data...
serializer.cpp Loading commit data...
serializer.hpp Loading commit data...
shape.cpp Loading commit data...
shape.hpp Loading commit data...
util.cpp Loading commit data...
util.hpp Loading commit data...
uuid.hpp Loading commit data...