{"id":"https://openalex.org/W4386272884","doi":"https://doi.org/10.48550/arxiv.2308.14364","title":"Target-independent XLA optimization using Reinforcement Learning","display_name":"Target-independent XLA optimization using Reinforcement Learning","publication_year":2023,"publication_date":"2023-08-28","ids":{"openalex":"https://openalex.org/W4386272884","doi":"https://doi.org/10.48550/arxiv.2308.14364"},"language":"en","primary_location":{"id":"pmh:oai:arXiv.org:2308.14364","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2308.14364","pdf_url":"https://arxiv.org/pdf/2308.14364","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},"type":"preprint","indexed_in":["arxiv","datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://arxiv.org/pdf/2308.14364","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5031297903","display_name":"Milan Ganai","orcid":"https://orcid.org/0009-0001-1825-3650"},"institutions":[],"countries":[],"is_corresponding":true,"raw_author_name":"Ganai, Milan","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5037312615","display_name":"Haichen Li","orcid":"https://orcid.org/0000-0002-7429-4633"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Li, Haichen","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5092715771","display_name":"Theodore Enns","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Enns, Theodore","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5101638215","display_name":"Yida Wang","orcid":"https://orcid.org/0000-0003-1350-485X"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Yida","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5074836221","display_name":"Randy Huang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Huang, Randy","raw_affiliation_strings":[],"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":5,"corresponding_author_ids":["https://openalex.org/A5031297903"],"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.9958000183105469,"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.9958000183105469,"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/T12535","display_name":"Machine Learning and Data Classification","score":0.9871000051498413,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"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/T12808","display_name":"Ferroelectric and Negative Capacitance Devices","score":0.9858999848365784,"subfield":{"id":"https://openalex.org/subfields/2208","display_name":"Electrical and Electronic Engineering"},"field":{"id":"https://openalex.org/fields/22","display_name":"Engineering"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/compiler","display_name":"Compiler","score":0.7773041725158691},{"id":"https://openalex.org/keywords/reinforcement-learning","display_name":"Reinforcement learning","score":0.7738263607025146},{"id":"https://openalex.org/keywords/computer-science","display_name":"Computer science","score":0.7697348594665527},{"id":"https://openalex.org/keywords/benchmark","display_name":"Benchmark (surveying)","score":0.7419862747192383},{"id":"https://openalex.org/keywords/kernel","display_name":"Kernel (algebra)","score":0.5083605647087097},{"id":"https://openalex.org/keywords/artificial-intelligence","display_name":"Artificial intelligence","score":0.44860202074050903},{"id":"https://openalex.org/keywords/domain","display_name":"Domain (mathematical analysis)","score":0.42548608779907227},{"id":"https://openalex.org/keywords/optimization-problem","display_name":"Optimization problem","score":0.4205196499824524},{"id":"https://openalex.org/keywords/deep-learning","display_name":"Deep learning","score":0.41123831272125244},{"id":"https://openalex.org/keywords/machine-learning","display_name":"Machine learning","score":0.3555721044540405},{"id":"https://openalex.org/keywords/theoretical-computer-science","display_name":"Theoretical computer science","score":0.3328116536140442},{"id":"https://openalex.org/keywords/algorithm","display_name":"Algorithm","score":0.21144765615463257},{"id":"https://openalex.org/keywords/programming-language","display_name":"Programming language","score":0.11694368720054626},{"id":"https://openalex.org/keywords/mathematics","display_name":"Mathematics","score":0.08482551574707031}],"concepts":[{"id":"https://openalex.org/C169590947","wikidata":"https://www.wikidata.org/wiki/Q47506","display_name":"Compiler","level":2,"score":0.7773041725158691},{"id":"https://openalex.org/C97541855","wikidata":"https://www.wikidata.org/wiki/Q830687","display_name":"Reinforcement learning","level":2,"score":0.7738263607025146},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7697348594665527},{"id":"https://openalex.org/C185798385","wikidata":"https://www.wikidata.org/wiki/Q1161707","display_name":"Benchmark (surveying)","level":2,"score":0.7419862747192383},{"id":"https://openalex.org/C74193536","wikidata":"https://www.wikidata.org/wiki/Q574844","display_name":"Kernel (algebra)","level":2,"score":0.5083605647087097},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.44860202074050903},{"id":"https://openalex.org/C36503486","wikidata":"https://www.wikidata.org/wiki/Q11235244","display_name":"Domain (mathematical analysis)","level":2,"score":0.42548608779907227},{"id":"https://openalex.org/C137836250","wikidata":"https://www.wikidata.org/wiki/Q984063","display_name":"Optimization problem","level":2,"score":0.4205196499824524},{"id":"https://openalex.org/C108583219","wikidata":"https://www.wikidata.org/wiki/Q197536","display_name":"Deep learning","level":2,"score":0.41123831272125244},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.3555721044540405},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.3328116536140442},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.21144765615463257},{"id":"https://openalex.org/C199360897","wikidata":"https://www.wikidata.org/wiki/Q9143","display_name":"Programming language","level":1,"score":0.11694368720054626},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.08482551574707031},{"id":"https://openalex.org/C13280743","wikidata":"https://www.wikidata.org/wiki/Q131089","display_name":"Geodesy","level":1,"score":0.0},{"id":"https://openalex.org/C114614502","wikidata":"https://www.wikidata.org/wiki/Q76592","display_name":"Combinatorics","level":1,"score":0.0},{"id":"https://openalex.org/C205649164","wikidata":"https://www.wikidata.org/wiki/Q1071","display_name":"Geography","level":0,"score":0.0},{"id":"https://openalex.org/C134306372","wikidata":"https://www.wikidata.org/wiki/Q7754","display_name":"Mathematical analysis","level":1,"score":0.0}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:oai:arXiv.org:2308.14364","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2308.14364","pdf_url":"https://arxiv.org/pdf/2308.14364","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},{"id":"doi:10.48550/arxiv.2308.14364","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2308.14364","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":"article"}],"best_oa_location":{"id":"pmh:oai:arXiv.org:2308.14364","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2308.14364","pdf_url":"https://arxiv.org/pdf/2308.14364","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"pdf":true,"grobid_xml":false},"content_urls":{"pdf":"https://content.openalex.org/works/W4386272884.pdf"},"referenced_works_count":0,"referenced_works":[],"related_works":["https://openalex.org/W2378211422","https://openalex.org/W2745001401","https://openalex.org/W4321353415","https://openalex.org/W3096456556","https://openalex.org/W4240253816","https://openalex.org/W2130974462","https://openalex.org/W2028665553","https://openalex.org/W2086519370","https://openalex.org/W972276598","https://openalex.org/W2087343574"],"abstract_inverted_index":{"An":[0],"important":[1],"challenge":[2],"in":[3,20,66,142,152],"Machine":[4],"Learning":[5,80],"compilers":[6],"like":[7],"XLA":[8,21,37,70,86,118],"is":[9,54,61],"multi-pass":[10],"optimization":[11,23,36,51],"and":[12,28,105,137,163,173],"analysis.":[13],"There":[14],"has":[15],"been":[16],"recent":[17],"interest":[18],"chiefly":[19],"target-dependent":[22,57],"on":[24,34,156,166],"the":[25,46,95,107,132,177,181],"graph-level,":[26],"subgraph-level,":[27],"kernel-level":[29],"phases.":[30],"We":[31,90,115],"specifically":[32],"focus":[33],"target-independent":[35],"HLO":[38,87],"pass":[39,67,88],"ordering:":[40],"our":[41,143],"approach":[42,179],"aims":[43],"at":[44],"finding":[45],"optimal":[47,85,102],"sequence":[48],"of":[49,149,159],"compiler":[50,133],"passes,":[52],"which":[53],"decoupled":[55],"from":[56],"optimization.":[58],"However,":[59],"there":[60],"little":[62],"domain":[63],"specific":[64],"study":[65],"ordering":[68],"for":[69,84,110,113,134],"HLO.":[71],"To":[72],"this":[73],"end,":[74],"we":[75,145],"propose":[76,92],"introducing":[77],"deep":[78,96],"Reinforcement":[79],"(RL)":[81],"based":[82],"search":[83,103],"ordering.":[89,185],"also":[91],"enhancements":[93],"to":[94,99,125,129],"RL":[97,127],"algorithms":[98,128],"further":[100],"improve":[101],"performance":[104],"open":[106],"research":[108],"direction":[109],"domain-specific":[111],"guidance":[112],"RL.":[114],"create":[116],"an":[117,147],"Gym":[119],"experimentation":[120,144],"framework":[121],"as":[122],"a":[123,157,167],"tool":[124],"enable":[126],"interact":[130],"with":[131],"passing":[135],"optimizations":[136],"thereby":[138],"train":[139],"agents.":[140],"Overall,":[141],"observe":[146],"average":[148],"$13.3\\%$":[150],"improvement":[151,165],"operation":[153],"count":[154],"reduction":[155],"benchmark":[158,169],"GPT-2":[160],"training":[161],"graphs":[162,175],"$10.4\\%$":[164],"diverse":[168],"including":[170],"GPT-2,":[171],"BERT,":[172],"ResNet":[174],"using":[176],"proposed":[178],"over":[180],"compiler's":[182],"default":[183],"phase":[184]},"counts_by_year":[],"updated_date":"2026-02-09T09:26:11.010843","created_date":"2025-10-10T00:00:00"}