mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

ActivationLeaky IpluginV2

Browse Source
This commit is contained in:
Micaela Verucchi
2021-06-28 23:23:15 +02:00
parent c306b36860
commit b86a93e85d
2 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CMakeLists.txt
+2 -2
View File
@@ -3,10 +3,10 @@ cmake_minimum_required(VERSION 3.15)
project (tkDNN) project (tkDNN)
set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} ${CMAKE_CURRENT_SOURCE_DIR}/cmake) set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} ${CMAKE_CURRENT_SOURCE_DIR}/cmake)
if(UNIX) if(UNIX)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -fPIC -Wno-deprecated-declarations -Wno-unused-variable ") set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++14 -fPIC -Wno-unused-variable ")
endif() endif()
if(WIN32) if(WIN32)
set(CMAKE_CXX_STANDARD 11) set(CMAKE_CXX_STANDARD 14)
set(CMAKE_CXX_FLAGS "/O2 /FS /EHsc") set(CMAKE_CXX_FLAGS "/O2 /FS /EHsc")
set(CMAKE_WINDOWS_EXPORT_ALL_SYMBOLS ON) set(CMAKE_WINDOWS_EXPORT_ALL_SYMBOLS ON)
endif(WIN32) endif(WIN32)
include/tkDNN/pluginsRT/ActivationLeakyRT.h
+14 -14
View File
@@ -1,7 +1,7 @@
#include<cassert> #include<cassert>
#include "../kernels.h" #include "../kernels.h"
class ActivationLeakyRT : public IPlugin { class ActivationLeakyRT : public IPluginV2 {
public: public:
ActivationLeakyRT() { ActivationLeakyRT() {
@@ -13,33 +13,33 @@ public:
} }
int getNbOutputs() const override { int getNbOutputs() const noexcept override {
return 1; return 1;
} }
Dims getOutputDimensions(int index, const Dims* inputs, int nbInputDims) override { Dims getOutputDimensions(int index, const Dims* inputs, int nbInputDims) noexcept override {
return inputs[0]; return inputs[0];
} }
void configure(const Dims* inputDims, int nbInputs, const Dims* outputDims, int nbOutputs, int maxBatchSize) override { //void configure(const Dims* inputDims, int nbInputs, const Dims* outputDims, int nbOutputs, int maxBatchSize) override {
size = 1; // size = 1;
for(int i=0; i<outputDims[0].nbDims; i++) // for(int i=0; i<outputDims[0].nbDims; i++)
size *= outputDims[0].d[i]; // size *= outputDims[0].d[i];
} //}
int initialize() override { int initialize() noexcept override {
return 0; return 0;
} }
virtual void terminate() override { virtual void terminate() noexcept override {
} }
virtual size_t getWorkspaceSize(int maxBatchSize) const override { virtual size_t getWorkspaceSize(int maxBatchSize) const noexcept override {
return 0; return 0;
} }
virtual int enqueue(int batchSize, const void*const * inputs, void** outputs, void* workspace, cudaStream_t stream) override { virtual int enqueue(int32_t batchSize, void const *const *inputs, void *const *outputs, void *workspace, cudaStream_t stream) noexcept override {
activationLEAKYForward((dnnType*)reinterpret_cast<const dnnType*>(inputs[0]), activationLEAKYForward((dnnType*)reinterpret_cast<const dnnType*>(inputs[0]),
reinterpret_cast<dnnType*>(outputs[0]), batchSize*size, stream); reinterpret_cast<dnnType*>(outputs[0]), batchSize*size, stream);
@@ -47,11 +47,11 @@ public:
} }
virtual size_t getSerializationSize() override { virtual size_t getSerializationSize() const noexcept override {
return 1*sizeof(int); return 1*sizeof(int);
} }
virtual void serialize(void* buffer) override { virtual void serialize(void* buffer) const noexcept override {
char *buf = reinterpret_cast<char*>(buffer),*a=buf; char *buf = reinterpret_cast<char*>(buffer),*a=buf;
tk::dnn::writeBUF(buf, size); tk::dnn::writeBUF(buf, size);
assert(buf == a + getSerializationSize()); assert(buf == a + getSerializationSize());