diff --git a/plugin/instanceNormalizationPlugin/instanceNormalizationPlugin.cu b/plugin/instanceNormalizationPlugin/instanceNormalizationPlugin.cu index f6a24995..48012e1c 100644 --- a/plugin/instanceNormalizationPlugin/instanceNormalizationPlugin.cu +++ b/plugin/instanceNormalizationPlugin/instanceNormalizationPlugin.cu @@ -69,7 +69,9 @@ InstanceNormalizationV3Plugin::InstanceNormalizationV3Plugin( for (int32_t c = 0; c < input.count; ++c) { auto const value = static_cast(input.values); - output.push_back(__internal_half2float(value[c])); + __half_raw raw; + raw.x = value[c]; + output.push_back(__half2float(raw)); } } else diff --git a/plugin/instanceNormalizationPlugin/instanceNormalizationPluginLegacy.cu b/plugin/instanceNormalizationPlugin/instanceNormalizationPluginLegacy.cu index 8f928818..c0c6badb 100644 --- a/plugin/instanceNormalizationPlugin/instanceNormalizationPluginLegacy.cu +++ b/plugin/instanceNormalizationPlugin/instanceNormalizationPluginLegacy.cu @@ -72,7 +72,9 @@ InstanceNormalizationPlugin::InstanceNormalizationPlugin( for (int32_t c = 0; c < input.count; ++c) { auto const value = static_cast(input.values); - output.push_back(__internal_half2float(value[c])); + __half_raw raw; + raw.x = value[c]; + output.push_back(__half2float(raw)); } } else