Fix includes and map assignment

triton-inference-server · Sep 28, 2023 · eb4ca9b · eb4ca9b
1 parent f255c22
commit eb4ca9b
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 5 deletions.
diff --git a/src/c++/library/common.h b/src/c++/library/common.h
@@ -34,11 +34,11 @@
 #include <functional>
 #include <iostream>
 #include <list>
-#include <map>
 #include <memory>
 #include <mutex>
 #include <string>
 #include <thread>
+#include <unordered_map>
 #include <vector>
 
 #ifdef TRITON_INFERENCE_SERVER_CLIENT_CLASS

diff --git a/src/c++/perf_analyzer/client_backend/triton/triton_client_backend.cc b/src/c++/perf_analyzer/client_backend/triton/triton_client_backend.cc
@@ -600,8 +600,7 @@ TritonClientBackend::ParseInferOptionsToTriton(
         static_cast<std::underlying_type<cb::RequestParameterType>::type>(
             map_entry.second.type);
     rp.type = static_cast<tc::RequestParameterType>(val);
-    triton_options->request_parameters.insert(
-        std::make_pair(map_entry.first, rp));
+    triton_options->request_parameters[map_entry.first] = rp;
   }
 }
 

diff --git a/src/c++/perf_analyzer/mock_infer_data_manager.h b/src/c++/perf_analyzer/mock_infer_data_manager.h
@@ -32,8 +32,6 @@
 
 namespace triton { namespace perfanalyzer {
 
-// std::unordered_map<std::string, clientbackend::RequestParameter>
-//     request_parameters;
 
 class MockInferDataManagerShm : public InferDataManagerShm {
  public: