Fixing compilation issue in embLayerNormPlugin (NVIDIA#248)

kvzhao · Dec 3, 2019 · fb305de · fb305de
1 parent d7dcaf1
commit fb305de
Showing 1 changed file with 1 addition and 1 deletion.
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);
 }