|
22 | 22 | #define DEPRECATED |
23 | 23 |
|
24 | 24 | class Dims { |
25 | | -public: |
26 | | - static const int32_t MAX_DIMS = 8; |
27 | | - int32_t nbDims; |
28 | | - int32_t d[MAX_DIMS]; |
| 25 | + public: |
| 26 | + static const int32_t MAX_DIMS = 8; |
| 27 | + int32_t nbDims; |
| 28 | + int32_t d[MAX_DIMS]; |
29 | 29 | }; |
30 | 30 |
|
31 | 31 | class Dims4 : public Dims { |
32 | | -public: |
33 | | - Dims4() { |
34 | | - nbDims = 4; |
35 | | - d[0] = d[1] = d[2] = d[3] = 0; |
36 | | - } |
37 | | - Dims4(int32_t d0, int32_t d1, |
38 | | - int32_t d2, int32_t d3) { |
39 | | - nbDims = 4; |
40 | | - d[0] = d0; |
41 | | - d[1] = d1; |
42 | | - d[2] = d2; |
43 | | - d[3] = d3; |
44 | | - } |
| 32 | + public: |
| 33 | + Dims4() { |
| 34 | + nbDims = 4; |
| 35 | + d[0] = d[1] = d[2] = d[3] = 0; |
| 36 | + } |
| 37 | + Dims4(int32_t d0, int32_t d1, int32_t d2, int32_t d3) { |
| 38 | + nbDims = 4; |
| 39 | + d[0] = d0; |
| 40 | + d[1] = d1; |
| 41 | + d[2] = d2; |
| 42 | + d[3] = d3; |
| 43 | + } |
45 | 44 | }; |
46 | 45 |
|
47 | 46 | class IPlugin { |
48 | | -public: |
49 | | - virtual int32_t getNbOutputs() const = 0; |
50 | | - virtual Dims getOutputDimensions(int32_t index, const Dims* inputs, |
51 | | - int32_t nbInputDims) = 0; |
52 | | - virtual void configure(const Dims* inputDims, int32_t nbInputs, |
53 | | - const Dims* outputDims, int32_t nbOutputs, |
54 | | - int32_t maxBatchSize) = 0; |
55 | | - virtual int32_t initialize() = 0; |
56 | | - virtual void terminate() = 0; |
57 | | - virtual size_t getWorkspaceSize(int32_t maxBatchSize) const = 0; |
58 | | - virtual int32_t enqueue(int32_t batchSize, const void* const* inputs, |
59 | | - void** outputs, void* workspace, |
60 | | - cudaStream_t stream) = 0; |
61 | | - virtual size_t getSerializationSize() = 0; |
62 | | - virtual void serialize(void* buffer) = 0; |
63 | | - virtual ~IPlugin() {} |
| 47 | + public: |
| 48 | + virtual int32_t getNbOutputs() const = 0; |
| 49 | + virtual Dims getOutputDimensions(int32_t index, const Dims* inputs, |
| 50 | + int32_t nbInputDims) = 0; |
| 51 | + virtual void configure(const Dims* inputDims, int32_t nbInputs, |
| 52 | + const Dims* outputDims, int32_t nbOutputs, |
| 53 | + int32_t maxBatchSize) = 0; |
| 54 | + virtual int32_t initialize() = 0; |
| 55 | + virtual void terminate() = 0; |
| 56 | + virtual size_t getWorkspaceSize(int32_t maxBatchSize) const = 0; |
| 57 | + virtual int32_t enqueue(int32_t batchSize, const void* const* inputs, |
| 58 | + void** outputs, void* workspace, |
| 59 | + cudaStream_t stream) = 0; |
| 60 | + virtual size_t getSerializationSize() = 0; |
| 61 | + virtual void serialize(void* buffer) = 0; |
| 62 | + virtual ~IPlugin() {} |
64 | 63 | }; |
65 | | - |
0 commit comments