Fix disagg with kvcache reuse for dsv32

Signed-off-by: Iman Tabrizian <10105175+tabrizian@users.noreply.github.com>

Author: Tabrizian

cpp/include/tensorrt_llm/batch_manager/kvCacheUtils.h

@@ -183,6 +183,10 @@ class BlockRange
             auto windowSize = cacheManager.getBlockManager().getPoolWindowSize(poolIdx);
             mPoolsPerWindow[windowSize].push_back(cacheManager.getBlockManager().getPrimaryPool(poolIdx));
         }
+        if (cacheManager.isEnableIndexerKCache())
+        {
+            mIndexerKCachePool = cacheManager.getIndexerKCachePool();
+        }
     }
 
     BlockRange(BaseKVCacheManager const& cacheManager, LlmRequest::RequestIdType requestId)

cpp/tensorrt_llm/batch_manager/dataTransceiver.cpp

@@ -806,7 +806,7 @@ class CacheReceiver::Impl
 
         RequestInfo requestInfo(requestId, mSelfState);
 
-        if (mFormatter->getCacheManager()->getBlockManager().getNumPools() == 1)
+        if (!mFormatter->getCacheManager()->getBlockManager().isVariableWindow())
         {
             auto* cacheManager = mFormatter->getCacheManager();
             auto beam = 0;