[WIP] Before fix needle.

This commit is contained in:
Zijie Tian
2025-12-31 23:35:25 +08:00
parent ccd1b3d4ab
commit 30462fe89a
5 changed files with 212 additions and 290 deletions

View File

@@ -29,10 +29,7 @@ def debug_load_hook(slot_idx: int, layer_id: int, cpu_block_id: int, k: Tensor,
"""Record loaded tensor values for layer 0."""
if layer_id != 0:
return
if layer_id == 0:
__import__('pdb').set_trace()
load_log.append({
"chunk_idx": current_chunk[0],
"cpu_block_id": cpu_block_id,