[HiCache] refactor: hicache normalization flow and compatibility checks (#19669)

2026-03-20 18:38:44 -07:00
parent 9419453713
commit 6c91590e1b
4 changed files with 184 additions and 46 deletions
@@ -728,6 +728,10 @@ class HiCacheController:
                return host_indices, device_indices.index_select(0, idx)
            elif self.mem_pool_host.layout == "page_first_direct":
                return host_indices, device_indices.cpu()
+            else:
+                raise ValueError(
+                    f"Unsupported layout {self.mem_pool_host.layout!r} for io backend 'direct'"
+                )
        elif self.io_backend == "kernel_ascend":
            return host_indices, device_indices.cpu()
        else: