[Relax][KV Cache] Refactor `_attention_sequence_prefill` function to …#17362

Merged

MasterJH5574 merged 1 commit intoapache:mainfrom

mengshyu:0910-kvcache

Sep 11, 2024

Commits on Sep 11, 2024

[Relax][KV Cache] Refactor `_attention_sequence_prefill` function to handle dynamic `batch_size` in TIR
mengshyu
committed