From f9e61da389a7f5f98d8ccbe33d4bab302bd1aca7 Mon Sep 17 00:00:00 2001
From: 1145284121 <1145284121@qq.com>
Date: Sat, 20 Sep 2025 16:31:05 +0800
Subject: [PATCH] [BugFix] fix image_ids bug in distributed + packed mode

---
 gsplat/rendering.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/gsplat/rendering.py b/gsplat/rendering.py
index 2054fed2e..ecebc7cd9 100644
--- a/gsplat/rendering.py
+++ b/gsplat/rendering.py
@@ -575,6 +575,9 @@ def reshape_view(C: int, world_view: torch.Tensor, N_world: list) -> torch.Tenso
                 output_splits=collected_splits,
             )
 
+            # image_ids must be updated after sparse_all_to_all, which is used in isect_tiles()
+            image_ids = camera_ids
+
             # Silently change C from global #Cameras to local #Cameras.
             C = C_world[world_rank]