Fix bug

triton-inference-server · Jun 1, 2023 · 6460553 · 6460553
1 parent eacf973
commit 6460553
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/rate_limiter.cc b/src/rate_limiter.cc
@@ -798,9 +798,9 @@ RateLimiter::ResourceManager::RemoveModelInstance(
     if (ditr != max_resources_.end()) {
       for (const auto& resource : resource_device_map.second) {
         auto ritr = ditr->second.find(resource.first);
-        if (ritr != ditr->second.end() && ritr->second >= resource.second) {
+        if (ritr != ditr->second.end() && ritr->second <= resource.second) {
           update_needed = true;
-          if (ritr->second > resource.second) {
+          if (ritr->second < resource.second) {
             LOG_ERROR << "Should not print this! Removing an instance with "
                          "resource above max resource.";
           }