Fix a bug that causes error when batchSize > 1 (#441)

zeyuznv · web-flow · commit 9d37f5df7309 · 2020-03-19T15:19:05.000-07:00
Signed-off-by: Jeffery Zhao &lt;zeyuz@nvidia.com&gt;
diff --git a/plugin/batchTilePlugin/batchTilePlugin.cpp b/plugin/batchTilePlugin/batchTilePlugin.cpp
@@ -95,7 +95,7 @@ DataType BatchTilePlugin::getOutputDataType(int index, const nvinfer1::DataType*
 
 int BatchTilePlugin::enqueue(int batchSize, const void* const* inputs, void** outputs, void*, cudaStream_t stream)
 {
-    float* output = reinterpret_cast<float*>(outputs[0]);
+    char* output = reinterpret_cast<char*>(outputs[0]);
     // expand to batch size
     for (int i = 0; i < batchSize; i++)
     {
@@ -210,4 +210,4 @@ IPluginV2Ext* BatchTilePluginCreator::createPlugin(const char* name, const Plugi
 IPluginV2Ext* BatchTilePluginCreator::deserializePlugin(const char* name, const void* serialData, size_t serialLength)
 {
     return new BatchTilePlugin(name, serialData, serialLength);
-}
+}

Original file line number	Diff line number	Diff line change
`@@ -95,7 +95,7 @@ DataType BatchTilePlugin::getOutputDataType(int index, const nvinfer1::DataType*`
`95`	`95`
`96`	`96`	`int BatchTilePlugin::enqueue(int batchSize, const void* const* inputs, void** outputs, void*, cudaStream_t stream)`
`97`	`97`	`{`
`98`		`- float* output = reinterpret_cast<float*>(outputs[0]);`
	`98`	`+ char* output = reinterpret_cast<char*>(outputs[0]);`
`99`	`99`	`// expand to batch size`
`100`	`100`	`for (int i = 0; i < batchSize; i++)`
`101`	`101`	`{`
`@@ -210,4 +210,4 @@ IPluginV2Ext* BatchTilePluginCreator::createPlugin(const char* name, const Plugi`
`210`	`210`	`IPluginV2Ext* BatchTilePluginCreator::deserializePlugin(const char* name, const void* serialData, size_t serialLength)`
`211`	`211`	`{`
`212`	`212`	`return new BatchTilePlugin(name, serialData, serialLength);`
`213`		`-}`
	`213`	`+}`