7abc574eeb
* 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> |
||
|---|---|---|
| .. | ||
| BarnesHutTsne.cpp | ||
| README.md | ||
| activations.cpp | ||
| addBias.cpp | ||
| adjust_hue.cpp | ||
| adjust_saturation.cpp | ||
| axis.cpp | ||
| batched_gemm.cpp | ||
| batchnorm.cpp | ||
| betaInc.cpp | ||
| col2im.cpp | ||
| compare_elem.cpp | ||
| confusion.cpp | ||
| convolutions.cpp | ||
| cross.cpp | ||
| d_t_s.cpp | ||
| diag.cpp | ||
| dilation2d.cpp | ||
| dropout.cpp | ||
| dynamic.cpp | ||
| extract_patches.cpp | ||
| fake_quantization.cpp | ||
| flatten.cpp | ||
| gather.cpp | ||
| gradient.cpp | ||
| gru.cpp | ||
| hamming.cpp | ||
| hashcode.cpp | ||
| histogram.cpp | ||
| histogramFixedWidth.cpp | ||
| im2col.cpp | ||
| image_resize.cpp | ||
| image_suppression.cpp | ||
| ismax.cpp | ||
| legacy_helper.cpp | ||
| lrn.cpp | ||
| lstm.cpp | ||
| lup.cpp | ||
| matrixSetDiag.cpp | ||
| matrix_band.cpp | ||
| matrix_diag_part.cpp | ||
| max_pooling.cpp | ||
| meshgrid.cpp | ||
| minimax.cpp | ||
| nth_element.cpp | ||
| one_hot.cpp | ||
| percentile.cpp | ||
| polyGamma.cpp | ||
| prefix.cpp | ||
| random_crop.cpp | ||
| range.cpp | ||
| reverse.cpp | ||
| roll.cpp | ||
| s_t_b.cpp | ||
| s_t_d.cpp | ||
| scatter.cpp | ||
| segment.cpp | ||
| sequence_mask.cpp | ||
| sg_cb.cpp | ||
| shift.cpp | ||
| sru.cpp | ||
| stack.cpp | ||
| svd.cpp | ||
| toggle_bits.cpp | ||
| top_k.cpp | ||
| transforms.cpp | ||
| weights.cpp | ||
| zeta.cpp | ||
README.md
This folder contains OpenMP implementations for operations helpers. Basically suited for homogenous x86-like platforms.