119 lines
3.9 KiB
C
119 lines
3.9 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 raver119 on 07.10.2017.
|
||
|
//
|
||
|
|
||
|
#ifndef LIBND4J_OPREGISTRATOR_H
|
||
|
#define LIBND4J_OPREGISTRATOR_H
|
||
|
|
||
|
#include <pointercast.h>
|
||
|
#include <vector>
|
||
|
#include <map>
|
||
|
#include <mutex>
|
||
|
#include <ops/declarable/DeclarableOp.h>
|
||
|
|
||
|
// handlers part
|
||
|
#include <cstdlib>
|
||
|
#include <csignal>
|
||
|
|
||
|
namespace nd4j {
|
||
|
namespace ops {
|
||
|
/**
|
||
|
* This class provides runtime ops lookup, based on opName or opHash.
|
||
|
* To build lookup directory we use *_OP_IMPL macro, which puts static structs at compile time in .cpp files,
|
||
|
* so once binary is executed, static objects are initialized automatically, and we get list of all ops
|
||
|
* available at runtime via this singleton.
|
||
|
*
|
||
|
*/
|
||
|
class ND4J_EXPORT OpRegistrator {
|
||
|
private:
|
||
|
static OpRegistrator* _INSTANCE;
|
||
|
OpRegistrator() {
|
||
|
nd4j_debug("OpRegistrator started\n","");
|
||
|
|
||
|
#ifndef _RELEASE
|
||
|
std::signal(SIGSEGV, &OpRegistrator::sigSegVHandler);
|
||
|
std::signal(SIGINT, &OpRegistrator::sigIntHandler);
|
||
|
std::signal(SIGABRT, &OpRegistrator::sigIntHandler);
|
||
|
std::signal(SIGFPE, &OpRegistrator::sigIntHandler);
|
||
|
std::signal(SIGILL, &OpRegistrator::sigIntHandler);
|
||
|
std::signal(SIGTERM, &OpRegistrator::sigIntHandler);
|
||
|
atexit(&OpRegistrator::exitHandler);
|
||
|
#endif
|
||
|
};
|
||
|
|
||
|
std::map<Nd4jLong, std::string> _msvc;
|
||
|
|
||
|
std::map<Nd4jLong, nd4j::ops::DeclarableOp*> _declarablesLD;
|
||
|
std::map<std::string, nd4j::ops::DeclarableOp*> _declarablesD;
|
||
|
std::vector<nd4j::ops::DeclarableOp *> _uniqueD;
|
||
|
|
||
|
std::mutex _locker;
|
||
|
std::string _opsList;
|
||
|
bool isInit = false;
|
||
|
public:
|
||
|
~OpRegistrator();
|
||
|
|
||
|
static OpRegistrator* getInstance();
|
||
|
|
||
|
static void exitHandler();
|
||
|
static void sigIntHandler(int sig);
|
||
|
static void sigSegVHandler(int sig);
|
||
|
|
||
|
void updateMSVC(Nd4jLong newHash, std::string& oldName);
|
||
|
|
||
|
template <typename T>
|
||
|
std::string local_to_string(T value);
|
||
|
const char * getAllCustomOperations();
|
||
|
|
||
|
/**
|
||
|
* This method registers operation
|
||
|
*
|
||
|
* @param op
|
||
|
*/
|
||
|
bool registerOperation(const char* name, nd4j::ops::DeclarableOp* op);
|
||
|
bool registerOperation(nd4j::ops::DeclarableOp *op);
|
||
|
|
||
|
nd4j::ops::DeclarableOp* getOperation(const char *name);
|
||
|
nd4j::ops::DeclarableOp* getOperation(Nd4jLong hash);
|
||
|
nd4j::ops::DeclarableOp* getOperation(std::string& name);
|
||
|
|
||
|
std::vector<Nd4jLong> getAllHashes();
|
||
|
|
||
|
int numberOfOperations();
|
||
|
};
|
||
|
|
||
|
|
||
|
/*
|
||
|
* These structs are used to "register" our ops in OpRegistrator.
|
||
|
*/
|
||
|
template <typename OpName>
|
||
|
struct __registrator{
|
||
|
__registrator();
|
||
|
};
|
||
|
|
||
|
template <typename OpName>
|
||
|
struct __registratorSynonym {
|
||
|
__registratorSynonym(const char *name, const char *oname);
|
||
|
};
|
||
|
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#endif //LIBND4J_OPREGISTRATOR_H
|