@@ -2959,6 +2959,10 @@ Status TensorrtExecutionProvider::CreateNodeComputeInfoFromGraph(const GraphView
2959
2959
}
2960
2960
2961
2961
// Force Pow + Reduce ops in layer norm to run in FP32 to avoid overflow
2962
+ #if defined(_MSC_VER)
2963
+ #pragma warning(push)
2964
+ #pragma warning(disable : 4996)
2965
+ #endif
2962
2966
if ((fp16_enable_ || bf16_enable_) && layer_norm_fp32_fallback_) {
2963
2967
for (auto idx = 1 ; idx < trt_network->getNbLayers () - 1 ; ++idx) {
2964
2968
auto layer = trt_network->getLayer (idx);
@@ -2972,6 +2976,9 @@ Status TensorrtExecutionProvider::CreateNodeComputeInfoFromGraph(const GraphView
2972
2976
}
2973
2977
}
2974
2978
}
2979
+ #if defined(_MSC_VER)
2980
+ #pragma warning(pop)
2981
+ #endif
2975
2982
2976
2983
int num_inputs = trt_network->getNbInputs ();
2977
2984
int num_outputs = trt_network->getNbOutputs ();
@@ -3146,6 +3153,10 @@ Status TensorrtExecutionProvider::CreateNodeComputeInfoFromGraph(const GraphView
3146
3153
}
3147
3154
}
3148
3155
3156
+ #if defined(_MSC_VER)
3157
+ #pragma warning(push)
3158
+ #pragma warning(disable : 4996)
3159
+ #endif
3149
3160
// Set precision flags
3150
3161
std::string trt_node_name_with_precision = fused_node.Name ();
3151
3162
if (fp16_enable_) {
@@ -3163,7 +3174,9 @@ Status TensorrtExecutionProvider::CreateNodeComputeInfoFromGraph(const GraphView
3163
3174
trt_node_name_with_precision += " _int8" ;
3164
3175
LOGS_DEFAULT (VERBOSE) << " [TensorRT EP] INT8 mode is enabled" ;
3165
3176
}
3166
-
3177
+ #if defined(_MSC_VER)
3178
+ #pragma warning(pop)
3179
+ #endif
3167
3180
// Set DLA
3168
3181
if (fp16_enable_ || int8_enable_) {
3169
3182
if (dla_enable_ && dla_core_ >= 0 ) { // DLA can only run with FP16 and INT8
@@ -3779,7 +3792,10 @@ Status TensorrtExecutionProvider::CreateNodeComputeInfoFromGraph(const GraphView
3779
3792
return ORT_MAKE_STATUS (ONNXRUNTIME, EP_FAIL, " TensorRT EP failed to set INT8 dynamic range." );
3780
3793
}
3781
3794
}
3782
-
3795
+ #if defined(_MSC_VER)
3796
+ #pragma warning(push)
3797
+ #pragma warning(disable : 4996)
3798
+ #endif
3783
3799
// Set precision
3784
3800
if (trt_state->int8_enable ) {
3785
3801
trt_config->setFlag (nvinfer1::BuilderFlag::kINT8 );
@@ -3793,7 +3809,9 @@ Status TensorrtExecutionProvider::CreateNodeComputeInfoFromGraph(const GraphView
3793
3809
trt_config->setFlag (nvinfer1::BuilderFlag::kBF16 );
3794
3810
LOGS_DEFAULT (VERBOSE) << " [TensorRT EP] BF16 mode is enabled" ;
3795
3811
}
3796
-
3812
+ #if defined(_MSC_VER)
3813
+ #pragma warning(pop)
3814
+ #endif
3797
3815
// Set DLA (DLA can only run with FP16 or INT8)
3798
3816
if ((trt_state->fp16_enable || trt_state->int8_enable ) && trt_state->dla_enable ) {
3799
3817
LOGS_DEFAULT (VERBOSE) << " [TensorRT EP] use DLA core " << trt_state->dla_core ;
0 commit comments