wehub/deeplearning4j--deeplearning4j
1
0
Fork 0
Code Issues Pull Requests Actions Deployments Models Agent Notes Packages Projects Releases Wiki Activity

chore: import upstream snapshot with attribution

Browse Source
This commit is contained in:
wehub-resource-sync
2026-07-13 12:47:05 +08:00
commit 4f3b7da785
7394 changed files with 2005594 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libnd4j/include/legacy/cpu/NativeOpExecutioner_scalar_tad_ints.cpp
+39
View File
@@ -0,0 +1,39 @@
/* ******************************************************************************
*
* Scalar TAD operations - INTEGER OUTPUT TYPES ONLY
*
******************************************************************************/
#include <system/op_boilerplate.h>
#include <array/DataTypeUtils.h>
#include <exceptions/datatype_exception.h>
#include <execution/Threads.h>
#include <legacy/NativeOpExecutioner.h>
#include <loops/scalar.h>
#include <types/types.h>
////////////////////////////////////////////////////////////////////////
// TAD execScalarInt - integer output types only (13 x 13 x 9 = 1,521 combinations)
void NativeOpExecutioner::execScalarInt(sd::LaunchContext *lc, int opNum, void const *hX, sd::LongType const *hXShapeInfo,
void const *dX, sd::LongType const *dXShapeInfo, void *extraParams, void *hZ,
sd::LongType const *hZShapeInfo, void *dZ, sd::LongType const *dZShapeInfo,
void const *hScalars, sd::LongType const *hScalarShapeInfo, void const *dScalars,
sd::LongType const *dScalarShapeInfo, sd::LongType *dimension, sd::LongType dimensionLength,
sd::LongType const *tadShapeInfo, sd::LongType const *tadOffsets,
sd::LongType const *tadShapeInfoZ, sd::LongType const *tadOffsetsZ) {
auto xType = sd::ArrayOptions::dataType(hXShapeInfo);
auto yType = sd::ArrayOptions::dataType(hScalarShapeInfo);
auto zType = sd::ArrayOptions::dataType(hZShapeInfo);
auto yLen = shape::length(hScalarShapeInfo);
auto func = PRAGMA_THREADS_FOR {
BUILD_TRIPLE_SELECTOR(xType, yType, zType, functions::scalar::ScalarTransform,
::transform(opNum, hX, hXShapeInfo, extraParams, hZ, hZShapeInfo, hScalars, dimension, dimensionLength,
tadShapeInfo, tadOffsets, tadShapeInfoZ, tadOffsetsZ, start, stop),
SD_COMMON_TYPES, SD_COMMON_TYPES, SD_INTEGER_TYPES);
};
samediff::Threads::parallel_tad(func, 0, yLen, 1, sd::math::sd_min<int>(yLen, sd::Environment::getInstance().maxMasterThreads()));
}