Fix includes and map assignment

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

Hide file tree

Showing 3 changed files with 2 additions and 3 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
@@ -602,6 +602,7 @@ TritonClientBackend::ParseInferOptionsToTriton(
     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: