029b84e2b7
* RL4J: Add generic update rule (#502) Signed-off-by: Alexandre Boulanger <aboulang2002@yahoo.com> * Shyrma reduce (#481) * - start working on improving of cpu legacy code for reduce ops Signed-off-by: Yurii <iuriish@yahoo.com> * - further work on improving legacy loops Signed-off-by: Yurii <iuriish@yahoo.com> * - still working on improving reduce ops Signed-off-by: Yurii <iuriish@yahoo.com> * - further work on improving reduce ops Signed-off-by: Yurii <iuriish@yahoo.com> * - testing speed run of new reduce op Signed-off-by: Yurii <iuriish@yahoo.com> * - working on improvement of default loop for reduce op Signed-off-by: Yurii <iuriish@yahoo.com> * - update signatures of stuff which calls reduce ops Signed-off-by: Yurii <iuriish@yahoo.com> * - make corrections in cuda reduce kernels Signed-off-by: Yurii <iuriish@yahoo.com> * - change loop for default case in broadcast legacy ops Signed-off-by: Yurii <iuriish@yahoo.com> * - comment some shape stuff Signed-off-by: Yurii <iuriish@yahoo.com> * - comment unnecessary prints in RNGtests Signed-off-by: Yurii <iuriish@yahoo.com> * - finish to resolve conflicts after master has been merged Signed-off-by: Yurii <iuriish@yahoo.com> * - get rid of some compilation mistakes of cuda stuff Signed-off-by: Yurii <iuriish@yahoo.com> * - minor changes Signed-off-by: Yurii <iuriish@yahoo.com> * - further search for bug causing crash on java test Signed-off-by: Yurii <iuriish@yahoo.com> * - add scalar case in reduce_ ... exec stuff Signed-off-by: Yurii <iuriish@yahoo.com> * - minor corrections in NAtiveOps.cu Signed-off-by: Yurii <iuriish@yahoo.com> * - add switch to scalar case execReduceXD functions Signed-off-by: Yurii <iuriish@yahoo.com> * - add support for vectors old shape in ConstantShapeHelper::createShapeInfoWithNoUnitiesForReduce Signed-off-by: Yurii <iuriish@yahoo.com> * - correct cuda mirrorPad Signed-off-by: Yurii <iuriish@yahoo.com> * - add support for vectors old shape in cuda createShapeInfoWithNoUnitiesForReduce Signed-off-by: Yurii <iuriish@yahoo.com> Co-authored-by: raver119 <raver119@gmail.com> * Add support for CUDA 11.0 (#492) * Add support for CUDA 11.0 * libnd4j tweaks for CUDA 11 Signed-off-by: raver119@gmail.com <raver119@gmail.com> * bindings update, again? Signed-off-by: raver119@gmail.com <raver119@gmail.com> * * Update versions of JavaCPP Presets for FFmpeg, OpenBLAS, and NumPy * update API to match CUDA 8 Signed-off-by: raver119@gmail.com <raver119@gmail.com> * * Update version of JavaCPP Presets for CPython * C++ updated for cuDNN 8.0 Signed-off-by: raver119@gmail.com <raver119@gmail.com> * one more test Signed-off-by: raver119@gmail.com <raver119@gmail.com> * one more test Signed-off-by: raver119@gmail.com <raver119@gmail.com> * one more test Signed-off-by: raver119@gmail.com <raver119@gmail.com> * 128-bit alignment for workspaces Signed-off-by: raver119@gmail.com <raver119@gmail.com> * change seed in 1 test Signed-off-by: raver119@gmail.com <raver119@gmail.com> * Fix dependecy duplication in python4j-parent pom * Fix group id for in python4j-numpy * few tests tweaked Signed-off-by: raver119@gmail.com <raver119@gmail.com> * Remove macosx-x86_64-gpu from nd4j-tests-tensorflow * few minor tweaks for IndexReduce Signed-off-by: raver119@gmail.com <raver119@gmail.com> * one test removed Signed-off-by: raver119@gmail.com <raver119@gmail.com> Co-authored-by: raver119@gmail.com <raver119@gmail.com> Co-authored-by: Serhii Shepel <9946053+sshepel@users.noreply.github.com> * RL4J: Add SyncTrainer and AgentLearnerBuilder for a few algorithms (#504) Signed-off-by: Alexandre Boulanger <aboulang2002@yahoo.com> Co-authored-by: Alexandre Boulanger <44292157+aboulang2002@users.noreply.github.com> Co-authored-by: Yurii Shyrma <iuriish@yahoo.com> Co-authored-by: raver119 <raver119@gmail.com> Co-authored-by: Serhii Shepel <9946053+sshepel@users.noreply.github.com>
676 lines
33 KiB
C++
676 lines
33 KiB
C++
/*******************************************************************************
|
|
* Copyright (c) 2015-2018 Skymind, Inc.
|
|
*
|
|
* This program and the accompanying materials are made available under the
|
|
* terms of the Apache License, Version 2.0 which is available at
|
|
* https://www.apache.org/licenses/LICENSE-2.0.
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
* License for the specific language governing permissions and limitations
|
|
* under the License.
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
******************************************************************************/
|
|
|
|
//
|
|
// Created by agibsonccc on 1/28/16.
|
|
//
|
|
|
|
#ifndef NATIVEOPERATIONS_NATIVEOPEXCUTIONER_H
|
|
#define NATIVEOPERATIONS_NATIVEOPEXCUTIONER_H
|
|
|
|
|
|
#include <types/types.h>
|
|
#include <system/dll.h>
|
|
#include <ops/specials.h>
|
|
#include <ops/specials_sparse.h>
|
|
#include <execution/LaunchContext.h>
|
|
#include <array/ArrayOptions.h>
|
|
|
|
/**
|
|
* Native op executioner:
|
|
*
|
|
*/
|
|
|
|
class ND4J_EXPORT NativeOpExecutioner {
|
|
public:
|
|
/**
|
|
*
|
|
* @param opNum
|
|
* @param x
|
|
* @param xShapeInfo
|
|
* @param extraParams
|
|
* @param result
|
|
* @param resultShapeInfo
|
|
*/
|
|
static void execIndexReduceScalar(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo);
|
|
|
|
/**
|
|
*
|
|
* @param opNum
|
|
* @param x
|
|
* @param xShapeInfo
|
|
* @param extraParamsVals
|
|
* @param y
|
|
* @param yShapeInfo
|
|
* @param result
|
|
* @param resultShapeInfoBuffer
|
|
* @param dimension
|
|
* @param dimensionLength
|
|
*/
|
|
static void execReduce3Scalar(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParamsVals,
|
|
const void *hY, const Nd4jLong *hYShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo);
|
|
|
|
|
|
/**
|
|
*
|
|
* @param opNum
|
|
* @param x
|
|
* @param xShapeInfo
|
|
* @param extraParamsVals
|
|
* @param y
|
|
* @param yShapeInfo
|
|
* @param result
|
|
* @param resultShapeInfo
|
|
*/
|
|
static void execReduce3(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParamsVals,
|
|
const void *hY, const Nd4jLong *hYShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo);
|
|
|
|
/**
|
|
*
|
|
* @param opNum
|
|
* @param x
|
|
* @param xShapeInfo
|
|
* @param extraParamsVals
|
|
* @param y
|
|
* @param yShapeInfo
|
|
* @param result
|
|
* @param resultShapeInfoBuffer
|
|
* @param dimension
|
|
* @param dimensionLength
|
|
*/
|
|
static void execReduce3(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParamsVals,
|
|
const void *hY, const Nd4jLong *hYShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
int *dimension, int dimensionLength,
|
|
const Nd4jLong *xTadOnlyShapeInfo, const Nd4jLong *xTadOffsets,
|
|
const Nd4jLong *yTadOnlyShapeInfo, const Nd4jLong *yTadOffsets);
|
|
|
|
static void execReduce3All(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParamsVals,
|
|
const void *hY, const Nd4jLong *hYShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
int *dimension, int dimensionLength,
|
|
const Nd4jLong *xTadShapeInfo, const Nd4jLong *xOffsets,
|
|
const Nd4jLong *yTadShapeInfo, const Nd4jLong *yOffsets);
|
|
|
|
/**
|
|
*
|
|
* @param opNum
|
|
* @param x
|
|
* @param xShapeInfo
|
|
* @param extraParams
|
|
* @param result
|
|
* @param resultShapeInfoBuffer
|
|
* @param dimension
|
|
* @param dimensionLength
|
|
*/
|
|
static void execIndexReduce(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
int *dimension, int dimensionLength,
|
|
const Nd4jLong *tadShapeInfo, const Nd4jLong *tadOffsets);
|
|
|
|
/**
|
|
*
|
|
* @param opNum
|
|
* @param x
|
|
* @param xStride
|
|
* @param result
|
|
* @param resultStride
|
|
* @param scalar
|
|
* @param extraParams
|
|
* @param n
|
|
*/
|
|
static void execScalar(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
const void *hScalar, const Nd4jLong *hSscalarShapeInfo,
|
|
const void *dScalar, const Nd4jLong *dSscalarShapeInfo,
|
|
void *extraParams,
|
|
bool allowParallelism = true);
|
|
|
|
static void execScalarBool(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
const void *hScalar, const Nd4jLong *hSscalarShapeInfo,
|
|
const void *dScalar, const Nd4jLong *dSscalarShapeInfo,
|
|
void *extraParams,
|
|
bool allowParallelism = true);
|
|
|
|
static void execScalarInt(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
const void *hScalar, const Nd4jLong *hSscalarShapeInfo,
|
|
const void *dScalar, const Nd4jLong *dSscalarShapeInfo,
|
|
void *extraParams,
|
|
bool allowParallelism = true);
|
|
|
|
static void execScalar(sd::LaunchContext *lc,
|
|
int opNum,
|
|
void const* hX, Nd4jLong const* hXShapeInfo,
|
|
void const* dX, Nd4jLong const* dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, Nd4jLong const* hZShapeInfo,
|
|
void *dZ, Nd4jLong const* dZShapeInfo,
|
|
void const* hScalars, Nd4jLong const* hScalarShapeInfo,
|
|
void const* dScalars, Nd4jLong const* dScalarShapeInfo,
|
|
int *dimension, int dimensionLength,
|
|
Nd4jLong const* tadShapeInfo, Nd4jLong const* tadOffsets,
|
|
Nd4jLong const* tadShapeInfoZ, Nd4jLong const* tadOffsetsZ);
|
|
|
|
static void execScalarBool(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
const void *hScalars, const Nd4jLong *hScalarShapeInfo,
|
|
const void *dScalars, const Nd4jLong *dScalarShapeInfo,
|
|
int *dimension, int dimensionLength,
|
|
const Nd4jLong *tadShapeInfo, const Nd4jLong *tadOffsets,
|
|
const Nd4jLong *tadShapeInfoZ, const Nd4jLong *tadOffsetsZ);
|
|
|
|
static void execScalarInt(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
const void *hScalars, const Nd4jLong *hScalarShapeInfo,
|
|
const void *dScalars, const Nd4jLong *dScalarShapeInfo,
|
|
int *dimension, int dimensionLength,
|
|
const Nd4jLong *tadShapeInfo, const Nd4jLong *tadOffsets,
|
|
const Nd4jLong *tadShapeInfoZ, const Nd4jLong *tadOffsetsZ);
|
|
|
|
|
|
/**
|
|
*
|
|
* @param opNum
|
|
* @param x
|
|
* @param xShapeInfo
|
|
* @param y
|
|
* @param yShapeInfo
|
|
* @param result
|
|
* @param resultShapeInfo
|
|
* @param dimension
|
|
* @param dimensionLength
|
|
*/
|
|
static void execBroadcast(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
const void *hY, const Nd4jLong *hYShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
int *dimension, int dimensionLength,
|
|
const Nd4jLong *tadOnlyShapeInfo, const Nd4jLong *tadOffsets,
|
|
const Nd4jLong *tadOnlyShapeInfoZ,const Nd4jLong *tadOffsetsZ);
|
|
|
|
static void execBroadcast(sd::LaunchContext* lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
const void *hY, const Nd4jLong *hYShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo);
|
|
|
|
static void execInverseBroadcast(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *x, const Nd4jLong *xShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
const void *y, const Nd4jLong *yShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *result, const Nd4jLong *resultShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
int *dimension, int dimensionLength,
|
|
const Nd4jLong *tadOnlyShapeInfo, const Nd4jLong *tadOffsets,
|
|
const Nd4jLong *tadOnlyShapeInfoZ, const Nd4jLong *tadOffsetsZ);
|
|
|
|
|
|
static void execBroadcastBool(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
const void *hY, const Nd4jLong *hYShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
void *extraParams,
|
|
int *dimension, int dimensionLength,
|
|
const Nd4jLong *tadOnlyShapeInfo, const Nd4jLong *tadOffsets,
|
|
const Nd4jLong *tadOnlyShapeInfoZ,const Nd4jLong *tadOffsetsZ);
|
|
|
|
static void execBroadcastBool(sd::LaunchContext* lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
const void *hY, const Nd4jLong *hYShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
void *extraParams);
|
|
|
|
static void execInverseBroadcastBool(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *x, const Nd4jLong *xShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
const void *y, const Nd4jLong *yShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *result, const Nd4jLong *resultShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
void *extraParams,
|
|
int *dimension, int dimensionLength,
|
|
const Nd4jLong *tadOnlyShapeInfo, const Nd4jLong *tadOffsets,
|
|
const Nd4jLong *tadOnlyShapeInfoZ, const Nd4jLong *tadOffsetsZ);
|
|
|
|
static void execBroadcastInt(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
const void *hY, const Nd4jLong *hYShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
int *dimension, int dimensionLength,
|
|
const Nd4jLong *tadOnlyShapeInfo, const Nd4jLong *tadOffsets,
|
|
const Nd4jLong *tadOnlyShapeInfoZ,const Nd4jLong *tadOffsetsZ);
|
|
|
|
static void execBroadcastInt(sd::LaunchContext* lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
const void *hY, const Nd4jLong *hYShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo);
|
|
|
|
static void execInverseBroadcastInt(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *x, const Nd4jLong *xShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
const void *y, const Nd4jLong *yShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *result, const Nd4jLong *resultShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
int *dimension, int dimensionLength,
|
|
const Nd4jLong *tadOnlyShapeInfo, const Nd4jLong *tadOffsets,
|
|
const Nd4jLong *tadOnlyShapeInfoZ, const Nd4jLong *tadOffsetsZ);
|
|
|
|
/**
|
|
*
|
|
* @param opNum
|
|
* @param dx
|
|
* @param xStride
|
|
* @param y
|
|
* @param yStride
|
|
* @param result
|
|
* @param resultStride
|
|
* @param extraParams
|
|
* @param n
|
|
*/
|
|
static void execPairwiseTransform(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
const void *hY, const Nd4jLong *hYShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
void *extraParams);
|
|
|
|
static void execPairwiseBoolTransform(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
const void *hY, const Nd4jLong *hYShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
void *extraParams);
|
|
|
|
static void execPairwiseIntTransform(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
const void *hY, const Nd4jLong *hYShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
void *extraParams);
|
|
|
|
/**
|
|
*
|
|
* @param opNum
|
|
* @param dx
|
|
* @param xStride
|
|
* @param result
|
|
* @param resultStride
|
|
* @param extraParams
|
|
* @param n
|
|
*/
|
|
static void execTransformFloat(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
void *extraParams,
|
|
const Nd4jLong *tadShapeInfo, const Nd4jLong *tadOffsets);
|
|
|
|
static void execTransformAny(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
void *extraParams,
|
|
const Nd4jLong *tadShapeInfo, const Nd4jLong *tadOffsets,
|
|
bool allowParallelism = true);
|
|
|
|
static void execTransformStrict(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
void *extraParams,
|
|
const Nd4jLong *tadShapeInfo, const Nd4jLong *tadOffsets);
|
|
|
|
static void execTransformSame(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
void *extraParams,
|
|
const Nd4jLong *tadShapeInfo, const Nd4jLong *tadOffsets);
|
|
|
|
static void execTransformBool(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
void *extraParams,
|
|
const Nd4jLong *tadShapeInfo, const Nd4jLong *tadOffsets);
|
|
/**
|
|
*
|
|
* @param opNum
|
|
* @param x
|
|
* @param xShapeInfo
|
|
* @param extraParams
|
|
* @param result
|
|
* @param resultShapeInfo
|
|
*/
|
|
static void execReduceFloat(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
int *dimension, int dimensionLength);
|
|
|
|
static void execReduceSame(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
int *dimension, int dimensionLength);
|
|
|
|
static void execReduceBool(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
int *dimension, int dimensionLength);
|
|
|
|
static void execReduceLong(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
int *dimension, int dimensionLength);
|
|
|
|
/**
|
|
*
|
|
* @param opNum
|
|
* @param x
|
|
* @param xShapeInfo
|
|
* @param extraParams
|
|
* @return
|
|
*/
|
|
static void execReduceFloatScalar(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo);
|
|
|
|
static void execReduceBoolScalar(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo);
|
|
|
|
static void execReduceSameScalar(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo);
|
|
|
|
static void execReduceLongScalar(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo);
|
|
|
|
static void execReduce3TAD(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParamsVals,
|
|
const void *hY, const Nd4jLong *hYShapeInfo,
|
|
const void *dY, const Nd4jLong *dYShapeInfo,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
int *dimension, int dimensionLength,
|
|
const Nd4jLong *tadShapeInfo, const Nd4jLong *tadOffsets,
|
|
const Nd4jLong *yTadShapeInfo, const Nd4jLong *yTadOffsets);
|
|
|
|
/**
|
|
*
|
|
* @param opNum
|
|
* @param x
|
|
* @param xShapeInfo
|
|
* @param extraParams
|
|
* @param result
|
|
* @param resultShapeInfoBuffer
|
|
* @param dimension
|
|
* @param dimensionLength
|
|
*/
|
|
static void execSummaryStats(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
int *dimension, int dimensionLength,
|
|
const Nd4jLong *tadShapeInfo, const Nd4jLong *tadOffsets,
|
|
bool biasCorrected);
|
|
|
|
/**
|
|
*
|
|
* @param opNum
|
|
* @param x
|
|
* @param xShapeInfo
|
|
* @param extraParams
|
|
* @param result
|
|
* @param resultShapeInfo
|
|
*/
|
|
static void execSummaryStats(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
bool biasCorrected);
|
|
|
|
/**
|
|
*
|
|
* @param opNum
|
|
* @param x
|
|
* @param xShapeInfo
|
|
* @param extraParams
|
|
* @param result
|
|
* @param resultShapeInfo
|
|
*/
|
|
static void execSummaryStatsScalar(sd::LaunchContext *lc,
|
|
int opNum,
|
|
const void *hX, const Nd4jLong *hXShapeInfo,
|
|
const void *dX, const Nd4jLong *dXShapeInfo,
|
|
void *extraParams,
|
|
void *hZ, const Nd4jLong *hZShapeInfo,
|
|
void *dZ, const Nd4jLong *dZShapeInfo,
|
|
bool biasCorrected);
|
|
|
|
|
|
static void execRandom(sd::LaunchContext *lc,
|
|
int opNum,
|
|
Nd4jPointer state,
|
|
void *hZ, const Nd4jLong *hZShapeBuffer,
|
|
void *dZ, const Nd4jLong *dZShapeBuffer,
|
|
void *extraArguments);
|
|
|
|
static void execRandom(sd::LaunchContext *lc,
|
|
int opNum,
|
|
Nd4jPointer state,
|
|
const void *hX, const Nd4jLong *hXShapeBuffer,
|
|
const void *dX, const Nd4jLong *dXShapeBuffer,
|
|
void *hZ, const Nd4jLong *hZShapeBuffer,
|
|
void *dZ, const Nd4jLong *dZShapeBuffer,
|
|
void *extraArguments);
|
|
|
|
static void execRandom(sd::LaunchContext *lc,
|
|
int opNum,
|
|
Nd4jPointer state,
|
|
const void *hX, const Nd4jLong *hXShapeBuffer,
|
|
const void *dX, const Nd4jLong *dXShapeBuffer,
|
|
const void *hY, const Nd4jLong *hYShapeBuffer,
|
|
const void *dY, const Nd4jLong *dYShapeBuffer,
|
|
void *hZ, const Nd4jLong *hZShapeBuffer,
|
|
void *dZ, const Nd4jLong *dZShapeBuffer,
|
|
void *extraArguments);
|
|
|
|
|
|
|
|
inline static void execSort(void *x, const Nd4jLong *xShapeInfo, bool descending) {
|
|
auto xType = sd::ArrayOptions::dataType(xShapeInfo);
|
|
|
|
BUILD_SINGLE_SELECTOR(xType, sd::SpecialMethods, ::sortGeneric(x, xShapeInfo, descending), LIBND4J_TYPES);
|
|
}
|
|
|
|
static void execSort(void *x, const Nd4jLong *xShapeInfo, int *dimension, int dimensionLength, const Nd4jLong *tadShapeInfo, const Nd4jLong *tadOffsets, bool descending) {
|
|
auto xType = sd::ArrayOptions::dataType(xShapeInfo);
|
|
|
|
BUILD_SINGLE_SELECTOR(xType, sd::SpecialMethods, ::sortTadGeneric(x, xShapeInfo, dimension, dimensionLength, tadShapeInfo, tadOffsets, descending), LIBND4J_TYPES);
|
|
}
|
|
|
|
inline static void execSortCooIndices(Nd4jLong *indices, void *values, Nd4jLong length, int rank) {
|
|
sd::sparse::SparseUtils<Nd4jLong>::sortCooIndicesGeneric(indices, reinterpret_cast<Nd4jLong *>(values), length, rank);
|
|
}
|
|
|
|
|
|
inline static Nd4jLong encodeBitmap(void *dx, const Nd4jLong *xShapeInfo, Nd4jLong N, int *dz, float threshold) {
|
|
auto xType = sd::ArrayOptions::dataType(xShapeInfo);
|
|
|
|
BUILD_SINGLE_SELECTOR(xType, return sd::SpecialMethods, ::encodeBitmapGeneric(dx, xShapeInfo, N, dz, threshold), FLOAT_TYPES);
|
|
}
|
|
|
|
inline static void decodeBitmap(const void *dx, Nd4jLong N, void *dz, const Nd4jLong *zShapeInfo) {
|
|
auto zType = sd::ArrayOptions::dataType(zShapeInfo);
|
|
|
|
BUILD_SINGLE_SELECTOR(zType, sd::SpecialMethods, ::decodeBitmapGeneric(dx, N, dz, zShapeInfo), FLOAT_TYPES);
|
|
}
|
|
|
|
};
|
|
|
|
|
|
#endif //NATIVEOPERATIONS_NATIVEOPEXCUTIONER_H
|