cavis/libnd4j/include/loops
raver119 7abc574eeb
Snapshot update (#8194)
* fix double consumption of rng on cpu

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

* Shyrma docs (#222)

* - documenting and profiling matrix_set_diag cuda kernel

Signed-off-by: Yurii <yurii@skymind.io>

* - correct formula of pnorm pooling in cuda 2d/3d kernels
- remove helper matrix_diag which duplicates work of helper matrix_set_diag

Signed-off-by: Yurii <yurii@skymind.io>

* cublasHandle sharing + lock

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

* cublasHandle sharing + lock

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

* Documentation from serialization/deserialization in NLP (#221)

* refactoring

Signed-off-by: Alexander Stoyakin <alexander.stoyakin@gmail.com>

* Javadocs

Signed-off-by: Alexander Stoyakin <alexander.stoyakin@gmail.com>

* Javadoc fixed

Signed-off-by: Alexander Stoyakin <alexander.stoyakin@gmail.com>

* Cleanup

Signed-off-by: Alexander Stoyakin <alexander.stoyakin@gmail.com>

* dedicated lock for getCudaCublasHandle

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

* Small fixes (#223)

Signed-off-by: AlexDBlack <blacka101@gmail.com>

* ELU DL4J fixes (#224)

Signed-off-by: AlexDBlack <blacka101@gmail.com>

* javadoc (#225)

Signed-off-by: Robert Altena <Rob@Ra-ai.com>

* Small test compilation fix (#226)

Signed-off-by: AlexDBlack <blacka101@gmail.com>

* #8182 remove spark version suffix (#227)

Signed-off-by: AlexDBlack <blacka101@gmail.com>

* [WIP] Thread safety (#229)

* sync after cublas*gemm

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

* mutex for CublasHelper

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

* don't store cublasHandle in LaunchContext, it's per-device anyway

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

* some printout

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

* check for field instead

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

* pew-pew

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

* don't release ContextBuffers until device changed

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

* small tweak

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

* some logging in sgemm

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

* stream sync

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

* some more logging

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

* some more error checks

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

* one fancy test

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

* one fancy test

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

* minor AffinityManager fix

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

* cudaEvent error logging improvement

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

* ConstantHelper thread safety

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

* - minor corrections in ConstantTadHelper

Signed-off-by: Yurii <yurii@skymind.io>

* ConstantShapeHelper thread safety

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

* ConstantTadHelper.cu updated

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

* logging off

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

* logging off

Signed-off-by: raver119 <raver119@gmail.com>
2019-09-03 22:02:02 +03:00
..
cpu Snapshot update (#8194) 2019-09-03 22:02:02 +03:00
cuda [WIP] minor (#218) 2019-09-02 11:25:48 +03:00
impl [WIP] multi-device support (#80) 2019-08-14 16:52:34 +03:00
BroadcastPairwiseConverter.h [WIP] Int broadcastables (#195) 2019-08-30 10:12:40 +03:00
BroadcastScalarConverter.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
ReduceType.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
aggregates.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
broadcasting.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
broadcasting_bool.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
broadcasting_int.h [WIP] Int broadcastables (#195) 2019-08-30 10:12:40 +03:00
grid_shaped.legacy Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
grid_strided.legacy Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
indexreduce.h [WIP] More tweaks (#173) 2019-08-27 10:37:10 +03:00
legacy_ops.h - add parameter alpha to elu and lrelu_bp (#213) 2019-08-31 20:57:39 +03:00
pairwise_bool.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
pairwise_int.h [WIP] Int broadcastables (#195) 2019-08-30 10:12:40 +03:00
pairwise_transform.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
random.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
reduce3.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
reduce_bool.h [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
reduce_float.h [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
reduce_long.h [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
reduce_same.h [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
scalar.h [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
scalar_bool.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
scalar_int.h [WIP] Int broadcastables (#195) 2019-08-30 10:12:40 +03:00
special_kernels.h [WIP] repeat op (#143) 2019-08-21 21:10:29 +03:00
summarystatsreduce.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
transform_any.h [WIP] More of CUDA (#95) 2019-08-05 11:27:05 +10:00
transform_bool.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
transform_float.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
transform_same.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
transform_strict.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
type_conversions.h [WIP] multi-device support (#80) 2019-08-14 16:52:34 +03:00