fix format

microsoft · Nov 24, 2023 · 04baad7 · 04baad7
1 parent 77a62f2
commit 04baad7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/onnxruntime/core/providers/tensorrt/onnx_ctx_model_helper.cc b/onnxruntime/core/providers/tensorrt/onnx_ctx_model_helper.cc
@@ -124,7 +124,7 @@ ONNX_NAMESPACE::ModelProto* CreateCtxNodeModel(const GraphViewer& graph_viewer,
   node_attributes->reserve(num_attributes);
   node_attributes->emplace(EMBED_MODE, *attr_0);
   node_attributes->emplace(EP_CACHE_CONTEXT, *attr_1);
-  
+
   if (compute_capability_enable) {
     cudaDeviceProp prop;
     CUDA_CALL_THROW(cudaGetDeviceProperties(&prop, device_id));

diff --git a/onnxruntime/core/providers/tensorrt/tensorrt_execution_provider.cc b/onnxruntime/core/providers/tensorrt/tensorrt_execution_provider.cc
@@ -3530,7 +3530,8 @@ Status TensorrtExecutionProvider::CreateNodeComputeFromGraph(const GraphViewer&
           LOGS_DEFAULT(VERBOSE) << "[TensorRT EP] Serialized timing cache " + timing_cache_path;
         }
       }
-
+
+      // dump ep context model
       if (dump_ep_context_model_ && ep_context_embed_mode_) {
         UpdateCtxNodeModelEngineContext(model_proto_.get(), reinterpret_cast<char*>(serialized_engine->data()), serialized_engine->size());
         DumpCtxNodeModel(model_proto_.get(), cache_path_prefix);