{"id":"https://openalex.org/W7162424328","doi":"https://doi.org/10.48550/arxiv.2605.23911","title":"Cross-Platform Fused MoE Dispatch in Triton: Portable Expert Routing Without CUDA","display_name":"Cross-Platform Fused MoE Dispatch in Triton: Portable Expert Routing Without CUDA","publication_year":2026,"publication_date":"2026-04-07","ids":{"openalex":"https://openalex.org/W7162424328","doi":"https://doi.org/10.48550/arxiv.2605.23911"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.23911","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.23911","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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.2605.23911","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5137055854","display_name":"Subhadip Mitra","orcid":null},"institutions":[],"countries":[],"is_corresponding":true,"raw_author_name":"Mitra, Subhadip","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":0,"corresponding_author_ids":["https://openalex.org/A5137055854"],"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/T10036","display_name":"Advanced Neural Network Applications","score":0.430400013923645,"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"}},"topics":[{"id":"https://openalex.org/T10036","display_name":"Advanced Neural Network Applications","score":0.430400013923645,"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/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.06019999831914902,"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/T14347","display_name":"Big Data and Digital Economy","score":0.04740000143647194,"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/cuda","display_name":"CUDA","score":0.6195999979972839},{"id":"https://openalex.org/keywords/correctness","display_name":"Correctness","score":0.5302000045776367},{"id":"https://openalex.org/keywords/router","display_name":"Router","score":0.46459999680519104},{"id":"https://openalex.org/keywords/code","display_name":"Code (set theory)","score":0.4449999928474426},{"id":"https://openalex.org/keywords/kernel","display_name":"Kernel (algebra)","score":0.4410000145435333},{"id":"https://openalex.org/keywords/inference","display_name":"Inference","score":0.44020000100135803},{"id":"https://openalex.org/keywords/skew","display_name":"Skew","score":0.42260000109672546},{"id":"https://openalex.org/keywords/inference-engine","display_name":"Inference engine","score":0.41499999165534973},{"id":"https://openalex.org/keywords/key","display_name":"Key (lock)","score":0.4072999954223633}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7759000062942505},{"id":"https://openalex.org/C2778119891","wikidata":"https://www.wikidata.org/wiki/Q477690","display_name":"CUDA","level":2,"score":0.6195999979972839},{"id":"https://openalex.org/C173608175","wikidata":"https://www.wikidata.org/wiki/Q232661","display_name":"Parallel computing","level":1,"score":0.5760999917984009},{"id":"https://openalex.org/C55439883","wikidata":"https://www.wikidata.org/wiki/Q360812","display_name":"Correctness","level":2,"score":0.5302000045776367},{"id":"https://openalex.org/C2775896111","wikidata":"https://www.wikidata.org/wiki/Q642560","display_name":"Router","level":2,"score":0.46459999680519104},{"id":"https://openalex.org/C2776760102","wikidata":"https://www.wikidata.org/wiki/Q5139990","display_name":"Code (set theory)","level":3,"score":0.4449999928474426},{"id":"https://openalex.org/C74193536","wikidata":"https://www.wikidata.org/wiki/Q574844","display_name":"Kernel (algebra)","level":2,"score":0.4410000145435333},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.44020000100135803},{"id":"https://openalex.org/C43711488","wikidata":"https://www.wikidata.org/wiki/Q7534783","display_name":"Skew","level":2,"score":0.42260000109672546},{"id":"https://openalex.org/C46743427","wikidata":"https://www.wikidata.org/wiki/Q1341685","display_name":"Inference engine","level":3,"score":0.41499999165534973},{"id":"https://openalex.org/C26517878","wikidata":"https://www.wikidata.org/wiki/Q228039","display_name":"Key (lock)","level":2,"score":0.4072999954223633},{"id":"https://openalex.org/C206729178","wikidata":"https://www.wikidata.org/wiki/Q2271896","display_name":"Scheduling (production processes)","level":2,"score":0.4065999984741211},{"id":"https://openalex.org/C74172769","wikidata":"https://www.wikidata.org/wiki/Q1446839","display_name":"Routing (electronic design automation)","level":2,"score":0.4058000147342682},{"id":"https://openalex.org/C160403385","wikidata":"https://www.wikidata.org/wiki/Q220543","display_name":"Queue","level":2,"score":0.3702000081539154},{"id":"https://openalex.org/C74912251","wikidata":"https://www.wikidata.org/wiki/Q6815727","display_name":"Memory footprint","level":2,"score":0.3653999865055084},{"id":"https://openalex.org/C115537543","wikidata":"https://www.wikidata.org/wiki/Q165596","display_name":"Cache","level":2,"score":0.34950000047683716},{"id":"https://openalex.org/C157764524","wikidata":"https://www.wikidata.org/wiki/Q1383412","display_name":"Throughput","level":3,"score":0.3431999981403351},{"id":"https://openalex.org/C113775141","wikidata":"https://www.wikidata.org/wiki/Q428691","display_name":"Computer engineering","level":1,"score":0.33649998903274536},{"id":"https://openalex.org/C149635348","wikidata":"https://www.wikidata.org/wiki/Q193040","display_name":"Embedded system","level":1,"score":0.32019999623298645},{"id":"https://openalex.org/C133162039","wikidata":"https://www.wikidata.org/wiki/Q1061077","display_name":"Code generation","level":3,"score":0.30219998955726624},{"id":"https://openalex.org/C46681722","wikidata":"https://www.wikidata.org/wiki/Q4725589","display_name":"Alias","level":2,"score":0.30079999566078186},{"id":"https://openalex.org/C43126263","wikidata":"https://www.wikidata.org/wiki/Q128751","display_name":"Source code","level":2,"score":0.28600001335144043},{"id":"https://openalex.org/C158379750","wikidata":"https://www.wikidata.org/wiki/Q214111","display_name":"Network packet","level":2,"score":0.2840999960899353},{"id":"https://openalex.org/C50630238","wikidata":"https://www.wikidata.org/wiki/Q971505","display_name":"General-purpose computing on graphics processing units","level":3,"score":0.2777999937534332},{"id":"https://openalex.org/C106251023","wikidata":"https://www.wikidata.org/wiki/Q851989","display_name":"Porting","level":3,"score":0.27070000767707825},{"id":"https://openalex.org/C2776303644","wikidata":"https://www.wikidata.org/wiki/Q1020499","display_name":"Interfacing","level":2,"score":0.2660999894142151},{"id":"https://openalex.org/C553261973","wikidata":"https://www.wikidata.org/wiki/Q14579","display_name":"Linux kernel","level":2,"score":0.2623000144958496},{"id":"https://openalex.org/C79403827","wikidata":"https://www.wikidata.org/wiki/Q3988","display_name":"Real-time computing","level":1,"score":0.2572000026702881},{"id":"https://openalex.org/C37404715","wikidata":"https://www.wikidata.org/wiki/Q380679","display_name":"Dynamic programming","level":2,"score":0.2538999915122986}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.23911","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.23911","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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.2605.23911","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.23911","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"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":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Mixture-of-Experts":[0],"(MoE)":[1],"architectures":[2],"power":[3],"the":[4,56,114,117,165],"majority":[5],"of":[6,102,113,116],"frontier":[7],"large":[8],"language":[9],"models,":[10],"but":[11],"their":[12],"inference":[13,121],"is":[14,80,190],"bottlenecked":[15],"by":[16],"irregular":[17],"memory":[18,104],"access":[19],"patterns":[20],"and":[21,36,67,130,142,163],"expert":[22,65,161],"routing":[23,157],"overhead.":[24],"Existing":[25],"optimized":[26],"MoE":[27,46],"kernels":[28],"(Megablocks,":[29],"Tutel,":[30],"FasterMoE)":[31],"are":[32],"implemented":[33],"in":[34,51],"CUDA":[35],"locked":[37],"to":[38,156],"NVIDIA":[39,108,140],"hardware.":[40],"We":[41,152],"present":[42],"TritonMoE,":[43],"a":[44,81],"fused":[45,82],"dispatch":[47],"kernel":[48,85],"written":[49],"entirely":[50],"OpenAI":[52],"Triton":[53,75],"that":[54,86],"performs":[55],"complete":[57],"forward":[58],"pass":[59,137],"--":[60,71,167,173],"router":[61],"scoring,":[62],"token":[63],"permutation,":[64],"GEMMs,":[66],"weighted":[68],"output":[69],"combination":[70],"using":[72],"only":[73],"portable":[74],"primitives.":[76],"Our":[77],"key":[78],"optimization":[79],"gate+up":[83],"GEMM":[84],"computes":[87],"both":[88,139],"SwiGLU":[89],"projections":[90],"from":[91],"shared":[92],"L2-cached":[93],"input":[94],"tiles":[95],"with":[96,145],"in-register":[97],"SiLU":[98],"activation,":[99],"eliminating":[100],"35%":[101],"global":[103],"traffic.":[105],"On":[106],"an":[107],"A100,":[109],"TritonMoE":[110],"achieves":[111],"89-131%":[112],"throughput":[115],"CUDA-optimized":[118],"Megablocks":[119],"at":[120,192],"batch":[122],"sizes":[123],"(<=":[124],"512":[125],"tokens)":[126],"across":[127],"Mixtral-8x7B,":[128],"DeepSeek-V3,":[129],"Qwen2-MoE":[131],"configurations.":[132],"All":[133],"162":[134],"correctness":[135],"tests":[136],"on":[138],"A100":[141],"AMD":[143],"MI300X":[144],"zero":[146],"code":[147],"changes,":[148],"validating":[149],"cross-platform":[150],"portability.":[151],"additionally":[153],"characterize":[154],"sensitivity":[155],"imbalance":[158],"under":[159,170],"Zipfian-skewed":[160],"assignments":[162],"identify":[164],"regime":[166],"64+":[168],"experts":[169],"extreme":[171],"skew":[172],"where":[174],"our":[175],"fixed-tile":[176],"scheduling":[177],"underperforms":[178],"Megablocks'":[179],"block-sparse":[180],"layout,":[181],"motivating":[182],"dynamic":[183],"block-to-expert":[184],"assignment":[185],"as":[186],"future":[187],"work.":[188],"Code":[189],"available":[191],"https://github.com/bassrehab/triton-kernels.":[193]},"counts_by_year":[],"updated_date":"2026-07-28T07:46:37.118299","created_date":"2026-05-27T00:00:00"}