fbcode//torchrec/inference:batching_queue_hipify_gen (meta-pytorch#3458)

kausv · meta-codesync[bot] · commit 24bb8480f3e5 · 2025-10-15T10:22:51.000-07:00
Summary: Pull Request resolved: meta-pytorch#3458 Pull Request resolved: meta-pytorch#3457 Reviewed By: nipung90 Differential Revision: D84599808 fbshipit-source-id: bcfacf2656f08b2f025c096d5c0947d5764605b7
diff --git a/torchrec/inference/inference_legacy/src/BatchingQueue.cpp b/torchrec/inference/inference_legacy/src/BatchingQueue.cpp
@@ -60,7 +60,7 @@ BatchingQueue::BatchingQueue(
       resourceManager_(std::move(resourceManager)) {
   CHECK(observer_ != nullptr);
   for (const auto& [_, metadata] : config_.batchingMetadata) {
-    if (batchingFuncs_.count(metadata.type) > 0) {
+    if (batchingFuncs_.contains(metadata.type)) {
       continue;
     }
     batchingFuncs_[metadata.type] =

Original file line number	Diff line number	Diff line change
`@@ -60,7 +60,7 @@ BatchingQueue::BatchingQueue(`
`60`	`60`	`resourceManager_(std::move(resourceManager)) {`
`61`	`61`	`CHECK(observer_ != nullptr);`
`62`	`62`	`for (const auto& [_, metadata] : config_.batchingMetadata) {`
`63`		`- if (batchingFuncs_.count(metadata.type) > 0) {`
	`63`	`+ if (batchingFuncs_.contains(metadata.type)) {`
`64`	`64`	`continue;`
`65`	`65`	`}`
`66`	`66`	`batchingFuncs_[metadata.type] =`