cavis/libnd4j/blas
raver119 269d508ba5
[WIP] cross-device migrations (#134)
* two more tests fixed

Signed-off-by: raver119 <raver119@gmail.com>

* CUDA device afinity tweaks

Signed-off-by: raver119 <raver119@gmail.com>

* minor tweaks

Signed-off-by: raver119 <raver119@gmail.com>

* prepareAction/registerAction for CustomOps

Signed-off-by: raver119 <raver119@gmail.com>

* lazy allocate host bufer before relocation

Signed-off-by: raver119 <raver119@gmail.com>

* one special test for migration in cpp

Signed-off-by: raver119 <raver119@gmail.com>

* tests update for msvc

Signed-off-by: raver119 <raver119@gmail.com>

* logging

Signed-off-by: raver119 <raver119@gmail.com>

* stick to old col2im impl

Signed-off-by: raver119 <raver119@gmail.com>

* cudaStreams reorganization

Signed-off-by: raver119 <raver119@gmail.com>

* buffer size fix

Signed-off-by: raver119 <raver119@gmail.com>

* c++ data migration

Signed-off-by: raver119 <raver119@gmail.com>

* fix CropAndResize test

Signed-off-by: raver119 <raver119@gmail.com>

* - minor improvment

Signed-off-by: Yurii <yurii@skymind.io>
2019-08-20 18:52:41 +03:00
..
cpu Shugeo cuda tests (#116) 2019-08-15 13:54:47 +03:00
cuda [WIP] cross-device migrations (#134) 2019-08-20 18:52:41 +03:00
CMakeLists.txt [WIP] multi-device support (#80) 2019-08-14 16:52:34 +03:00
Environment.cpp [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
Environment.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
GraphExecutioner.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
NDArray.h [WIP] cross-device migrations (#134) 2019-08-20 18:52:41 +03:00
NDArray.hpp [WIP] bitwise ops (#115) 2019-08-15 11:49:50 +03:00
NDArrayFactory.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
NativeOpExecutioner.h [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
NativeOps.h [WIP] multi-device support (#80) 2019-08-14 16:52:34 +03:00