Static IO manager bug fix (#12718)

limintang · facebook-github-bot · commit 6fa09360cd06 · 2025-07-22T11:28:25.000-07:00
Summary:

As title.

Reviewed By: sxu

Differential Revision: D78717294
diff --git a/examples/models/llama/runner/static_attention_io_manager.h b/examples/models/llama/runner/static_attention_io_manager.h
@@ -903,7 +903,7 @@ class StaticAttentionIOManager {
   }
 
   StaticAttentionIOConfig config_;
-  size_t input_pos_;
+  size_t input_pos_ = 0;
   StaticKVCache<CacheT, CacheAllocatorT> kCaches_;
   StaticKVCache<CacheT, CacheAllocatorT> vCaches_;
   std::unordered_map<size_t, StaticAttentionMask<MaskT, MaskAllocatorT>>

Original file line number	Diff line number	Diff line change
`@@ -903,7 +903,7 @@ class StaticAttentionIOManager {`
`903`	`903`	`}`
`904`	`904`
`905`	`905`	`StaticAttentionIOConfig config_;`
`906`		`- size_t input_pos_;`
	`906`	`+ size_t input_pos_ = 0;`
`907`	`907`	`StaticKVCache<CacheT, CacheAllocatorT> kCaches_;`
`908`	`908`	`StaticKVCache<CacheT, CacheAllocatorT> vCaches_;`
`909`	`909`	`std::unordered_map<size_t, StaticAttentionMask<MaskT, MaskAllocatorT>>`