{"id":"https://openalex.org/W7165636259","doi":"https://doi.org/10.48550/arxiv.2606.22968","title":"MOCAP: Wafer-Scale-Chip-Oriented Memory-Orchestrated Chunked Pipelining Framework for Prefill-Only LLM Inference","display_name":"MOCAP: Wafer-Scale-Chip-Oriented Memory-Orchestrated Chunked Pipelining Framework for Prefill-Only LLM Inference","publication_year":2026,"publication_date":"2026-06-22","ids":{"openalex":"https://openalex.org/W7165636259","doi":"https://doi.org/10.48550/arxiv.2606.22968"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2606.22968","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.22968","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":null,"license_id":null,"version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://doi.org/10.48550/arxiv.2606.22968","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5139195816","display_name":"Zichuan Wang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Zichuan","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5089405414","display_name":"Huizheng Wang","orcid":"https://orcid.org/0000-0002-9763-8208"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Huizheng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133363812","display_name":"Y H Xiao","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Xiao, Yuheng","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5130171344","display_name":"Haonan Zuo","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zuo, Haonan","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5069838515","display_name":"Tingcha Wei","orcid":"https://orcid.org/0000-0001-7619-4313"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wei, Taiquan","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5139201578","display_name":"Jinyi Deng","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Deng, Jinyi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5139202141","display_name":"Chao Li","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Li, Chao","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5139131631","display_name":"Yang Hu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Hu, Yang","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5139206639","display_name":"Shouyi Yin","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yin, Shouyi","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":0,"corresponding_author_ids":[],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":null,"has_fulltext":false,"cited_by_count":0,"citation_normalized_percentile":null,"cited_by_percentile_year":null,"biblio":{"volume":null,"issue":null,"first_page":null,"last_page":null},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.44119998812675476,"subfield":{"id":"https://openalex.org/subfields/1708","display_name":"Hardware and Architecture"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},"topics":[{"id":"https://openalex.org/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.44119998812675476,"subfield":{"id":"https://openalex.org/subfields/1708","display_name":"Hardware and Architecture"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10036","display_name":"Advanced Neural Network Applications","score":0.22509999573230743,"subfield":{"id":"https://openalex.org/subfields/1707","display_name":"Computer Vision and Pattern Recognition"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10101","display_name":"Cloud Computing and Resource Management","score":0.03779999911785126,"subfield":{"id":"https://openalex.org/subfields/1710","display_name":"Information Systems"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/bottleneck","display_name":"Bottleneck","score":0.7163000106811523},{"id":"https://openalex.org/keywords/cache","display_name":"Cache","score":0.6876000165939331},{"id":"https://openalex.org/keywords/pipeline","display_name":"Pipeline (software)","score":0.6729000210762024},{"id":"https://openalex.org/keywords/latency","display_name":"Latency (audio)","score":0.5338000059127808},{"id":"https://openalex.org/keywords/sequence","display_name":"Sequence (biology)","score":0.5239999890327454},{"id":"https://openalex.org/keywords/inference","display_name":"Inference","score":0.430400013923645},{"id":"https://openalex.org/keywords/partition","display_name":"Partition (number theory)","score":0.41429999470710754}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.8202999830245972},{"id":"https://openalex.org/C2780513914","wikidata":"https://www.wikidata.org/wiki/Q18210350","display_name":"Bottleneck","level":2,"score":0.7163000106811523},{"id":"https://openalex.org/C115537543","wikidata":"https://www.wikidata.org/wiki/Q165596","display_name":"Cache","level":2,"score":0.6876000165939331},{"id":"https://openalex.org/C43521106","wikidata":"https://www.wikidata.org/wiki/Q2165493","display_name":"Pipeline (software)","level":2,"score":0.6729000210762024},{"id":"https://openalex.org/C173608175","wikidata":"https://www.wikidata.org/wiki/Q232661","display_name":"Parallel computing","level":1,"score":0.6284999847412109},{"id":"https://openalex.org/C82876162","wikidata":"https://www.wikidata.org/wiki/Q17096504","display_name":"Latency (audio)","level":2,"score":0.5338000059127808},{"id":"https://openalex.org/C2778112365","wikidata":"https://www.wikidata.org/wiki/Q3511065","display_name":"Sequence (biology)","level":2,"score":0.5239999890327454},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.430400013923645},{"id":"https://openalex.org/C42812","wikidata":"https://www.wikidata.org/wiki/Q1082910","display_name":"Partition (number theory)","level":2,"score":0.41429999470710754},{"id":"https://openalex.org/C2779960059","wikidata":"https://www.wikidata.org/wiki/Q7113681","display_name":"Overhead (engineering)","level":2,"score":0.3668999969959259},{"id":"https://openalex.org/C188198153","wikidata":"https://www.wikidata.org/wiki/Q1613840","display_name":"Limiting","level":2,"score":0.3441999852657318},{"id":"https://openalex.org/C157764524","wikidata":"https://www.wikidata.org/wiki/Q1383412","display_name":"Throughput","level":3,"score":0.3352999985218048},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.3116999864578247},{"id":"https://openalex.org/C2776257435","wikidata":"https://www.wikidata.org/wiki/Q1576430","display_name":"Bandwidth (computing)","level":2,"score":0.30649998784065247},{"id":"https://openalex.org/C45374587","wikidata":"https://www.wikidata.org/wiki/Q12525525","display_name":"Computation","level":2,"score":0.2791000008583069},{"id":"https://openalex.org/C188045654","wikidata":"https://www.wikidata.org/wiki/Q17148339","display_name":"Memory bandwidth","level":2,"score":0.2784999907016754},{"id":"https://openalex.org/C120314980","wikidata":"https://www.wikidata.org/wiki/Q180634","display_name":"Distributed computing","level":1,"score":0.2581000030040741},{"id":"https://openalex.org/C189783530","wikidata":"https://www.wikidata.org/wiki/Q352090","display_name":"CPU cache","level":3,"score":0.25699999928474426},{"id":"https://openalex.org/C46686674","wikidata":"https://www.wikidata.org/wiki/Q466303","display_name":"Boosting (machine learning)","level":2,"score":0.25189998745918274}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2606.22968","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.22968","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":null,"license_id":null,"version":null,"is_accepted":false,"is_published":null,"raw_source_name":null,"raw_type":"Preprint"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2606.22968","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2606.22968","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":null,"license_id":null,"version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Large":[0],"language":[1],"models":[2],"(LLMs)":[3],"are":[4],"increasingly":[5],"used":[6],"in":[7,76],"prefill-only":[8,150],"workloads,":[9],"where":[10],"end-to-end":[11,212],"latency":[12,135,213],"is":[13,63],"dominated":[14],"by":[15,165,228],"the":[16,114,174,223],"prefill":[17,62,89],"phase.":[18],"For":[19],"long-context":[20,88],"prefill,":[21],"communication":[22,48],"overhead":[23],"grows":[24],"with":[25,206,233],"sequence":[26,69,116,176,226],"length":[27,227],"and":[28,50,54,73,112,194,214],"quickly":[29],"becomes":[30],"a":[31,41,66,77,144],"bottleneck":[32],"on":[33,90,129,153,218],"conventional":[34],"GPU":[35],"systems,":[36],"making":[37],"wafer-scale":[38],"chips":[39],"(WSCs)":[40],"promising":[42],"substrate":[43],"due":[44],"to":[45,60,64,87,102,133,161,185,230],"their":[46],"high":[47],"bandwidth":[49],"large":[51],"aggregate":[52],"compute":[53],"memory":[55,110,163],"capacity.":[56],"A":[57],"natural":[58],"way":[59],"accelerate":[61],"partition":[65],"long":[67],"input":[68],"into":[70],"multiple":[71],"chunks":[72,98,120],"execute":[74],"them":[75],"finer-grained":[78],"pipeline":[79,106,170,199],"across":[80,97,105,169],"devices.":[81],"However,":[82],"directly":[83],"applying":[84],"this":[85],"idea":[86],"WSCs":[91],"remains":[92],"challenging.":[93],"First,":[94],"causal":[95],"dependency":[96],"causes":[99],"KV":[100,158,167,195],"cache":[101,168],"accumulate":[103],"unevenly":[104],"stages,":[107,171],"creating":[108],"severe":[109],"imbalance":[111,164],"limiting":[113],"feasible":[115,175],"length.":[117,177],"Second,":[118],"later":[119],"require":[121],"more":[122],"attention":[123],"computation":[124],"because":[125],"each":[126],"chunk":[127,187],"depends":[128],"preceding":[130],"chunks,":[131],"leading":[132],"chunk-level":[134],"imbalance.":[136],"To":[137],"address":[138],"these":[139],"challenges,":[140],"we":[141],"present":[142],"MOCAP,":[143],"memory-orchestrated":[145],"chunked":[146],"pipelining":[147],"framework":[148],"for":[149],"LLM":[151],"inference":[152],"WSCs.":[154],"MOCAP":[155,208,220],"introduces":[156],"Memory-Balanced":[157],"Reallocation":[159],"(MBKR)":[160],"alleviate":[162],"redistributing":[166],"thereby":[172],"extending":[173],"It":[178],"further":[179],"incorporates":[180],"Latency-Balanced":[181],"Chunk":[182],"Partitioning":[183],"(LBCP)":[184],"balance":[186],"execution":[188],"cost":[189],"under":[190],"both":[191],"attention-cost":[192],"growth":[193],"reallocation":[196],"overhead,":[197],"improving":[198],"efficiency.":[200],"Experimental":[201],"results":[202],"show":[203],"that,":[204],"compared":[205,232],"GPipe,":[207],"achieves":[209],"76.4\\%":[210],"lower":[211],"3.24$\\times$":[215],"higher":[216],"throughput":[217],"average.":[219],"also":[221],"extends":[222],"maximum":[224],"supported":[225],"up":[229],"1.31$\\times$":[231],"Terapipe.":[234]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-06-24T00:00:00"}