Fixing compilation issue in embLayerNormPlugin (#248)

kevinch-nv · web-flow · commit fb305dea28a8 · 2019-12-03T10:45:49.000-08:00
diff --git a/plugin/embLayerNormPlugin/embLayerNormPlugin.cu b/plugin/embLayerNormPlugin/embLayerNormPlugin.cu
@@ -376,7 +376,7 @@ void EmbLayerNormPluginDynamic::configurePlugin(
     assert(inputs[0].desc.type == DataType::kINT32);
     assert(inputs[1].desc.type == DataType::kINT32);
     assert(inputs[2].desc.type == DataType::kINT32);
-    assert(outputs[0].desc.type == DataType::kFLOAT || out_type == DataType::kHALF);
+    assert(outputs[0].desc.type == DataType::kFLOAT || outputs[0].desc.type == DataType::kHALF);
     assert(outputs[1].desc.type == DataType::kINT32);
 }
 

Original file line number	Diff line number	Diff line change
`@@ -376,7 +376,7 @@ void EmbLayerNormPluginDynamic::configurePlugin(`
`376`	`376`	`assert(inputs[0].desc.type == DataType::kINT32);`
`377`	`377`	`assert(inputs[1].desc.type == DataType::kINT32);`
`378`	`378`	`assert(inputs[2].desc.type == DataType::kINT32);`
`379`		`- assert(outputs[0].desc.type == DataType::kFLOAT \|\| out_type == DataType::kHALF);`
	`379`	`+ assert(outputs[0].desc.type == DataType::kFLOAT \|\| outputs[0].desc.type == DataType::kHALF);`
`380`	`380`	`assert(outputs[1].desc.type == DataType::kINT32);`
`381`	`381`	`}`
`382`	`382`