{"id":"https://openalex.org/W4318541538","doi":"https://doi.org/10.1145/3575693.3576933","title":"TensorIR: An Abstraction for Automatic Tensorized Program Optimization","display_name":"TensorIR: An Abstraction for Automatic Tensorized Program Optimization","publication_year":2023,"publication_date":"2023-01-27","ids":{"openalex":"https://openalex.org/W4318541538","doi":"https://doi.org/10.1145/3575693.3576933"},"language":"en","primary_location":{"id":"doi:10.1145/3575693.3576933","is_oa":true,"landing_page_url":"https://doi.org/10.1145/3575693.3576933","pdf_url":"https://dl.acm.org/doi/pdf/10.1145/3575693.3576933","source":null,"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the 28th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 2","raw_type":"proceedings-article"},"type":"article","indexed_in":["crossref"],"open_access":{"is_oa":true,"oa_status":"gold","oa_url":"https://dl.acm.org/doi/pdf/10.1145/3575693.3576933","any_repository_has_fulltext":null},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5015218854","display_name":"Siyuan Feng","orcid":"https://orcid.org/0000-0002-3194-1124"},"institutions":[{"id":"https://openalex.org/I183067930","display_name":"Shanghai Jiao Tong University","ror":"https://ror.org/0220qvk04","country_code":"CN","type":"education","lineage":["https://openalex.org/I183067930"]}],"countries":["CN"],"is_corresponding":true,"raw_author_name":"Siyuan Feng","raw_affiliation_strings":["Shanghai Jiao Tong University, China"],"affiliations":[{"raw_affiliation_string":"Shanghai Jiao Tong University, China","institution_ids":["https://openalex.org/I183067930"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5065994853","display_name":"Bohan Hou","orcid":"https://orcid.org/0000-0001-5718-3387"},"institutions":[{"id":"https://openalex.org/I74973139","display_name":"Carnegie Mellon University","ror":"https://ror.org/05x2bcf33","country_code":"US","type":"education","lineage":["https://openalex.org/I74973139"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Bohan Hou","raw_affiliation_strings":["Carnegie Mellon University, USA"],"affiliations":[{"raw_affiliation_string":"Carnegie Mellon University, USA","institution_ids":["https://openalex.org/I74973139"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5101264780","display_name":"Hongyi Jin","orcid":"https://orcid.org/0000-0001-6894-6554"},"institutions":[{"id":"https://openalex.org/I74973139","display_name":"Carnegie Mellon University","ror":"https://ror.org/05x2bcf33","country_code":"US","type":"education","lineage":["https://openalex.org/I74973139"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Hongyi Jin","raw_affiliation_strings":["Carnegie Mellon University, USA"],"affiliations":[{"raw_affiliation_string":"Carnegie Mellon University, USA","institution_ids":["https://openalex.org/I74973139"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5076540989","display_name":"Wuwei Lin","orcid":"https://orcid.org/0000-0001-8589-5453"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wuwei Lin","raw_affiliation_strings":["OctoML, USA"],"affiliations":[{"raw_affiliation_string":"OctoML, USA","institution_ids":[]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5022832599","display_name":"Junru Shao","orcid":"https://orcid.org/0000-0002-7370-1495"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Junru Shao","raw_affiliation_strings":["OctoML, USA"],"affiliations":[{"raw_affiliation_string":"OctoML, USA","institution_ids":[]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5025482495","display_name":"Ruihang Lai","orcid":"https://orcid.org/0000-0001-6400-5079"},"institutions":[{"id":"https://openalex.org/I74973139","display_name":"Carnegie Mellon University","ror":"https://ror.org/05x2bcf33","country_code":"US","type":"education","lineage":["https://openalex.org/I74973139"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Ruihang Lai","raw_affiliation_strings":["Carnegie Mellon University, USA"],"affiliations":[{"raw_affiliation_string":"Carnegie Mellon University, USA","institution_ids":["https://openalex.org/I74973139"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5101740226","display_name":"Zihao Ye","orcid":"https://orcid.org/0000-0002-6450-8108"},"institutions":[{"id":"https://openalex.org/I201448701","display_name":"University of Washington","ror":"https://ror.org/00cvxb145","country_code":"US","type":"education","lineage":["https://openalex.org/I201448701"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Zihao Ye","raw_affiliation_strings":["University of Washington, USA"],"affiliations":[{"raw_affiliation_string":"University of Washington, USA","institution_ids":["https://openalex.org/I201448701"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5061339425","display_name":"Lianmin Zheng","orcid":"https://orcid.org/0000-0001-5812-731X"},"institutions":[{"id":"https://openalex.org/I95457486","display_name":"University of California, Berkeley","ror":"https://ror.org/01an7q238","country_code":"US","type":"education","lineage":["https://openalex.org/I95457486"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Lianmin Zheng","raw_affiliation_strings":["University of California at Berkeley, USA"],"affiliations":[{"raw_affiliation_string":"University of California at Berkeley, USA","institution_ids":["https://openalex.org/I95457486"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5008633310","display_name":"Cody Hao Yu","orcid":"https://orcid.org/0000-0002-9298-6254"},"institutions":[{"id":"https://openalex.org/I1311688040","display_name":"Amazon (United States)","ror":"https://ror.org/04mv4n011","country_code":"US","type":"company","lineage":["https://openalex.org/I1311688040"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Cody Hao Yu","raw_affiliation_strings":["Amazon Web Services, USA"],"affiliations":[{"raw_affiliation_string":"Amazon Web Services, USA","institution_ids":["https://openalex.org/I1311688040"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5001571390","display_name":"Yong Yu","orcid":"https://orcid.org/0000-0003-0281-8271"},"institutions":[{"id":"https://openalex.org/I183067930","display_name":"Shanghai Jiao Tong University","ror":"https://ror.org/0220qvk04","country_code":"CN","type":"education","lineage":["https://openalex.org/I183067930"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Yong Yu","raw_affiliation_strings":["Shanghai Jiao Tong University, China"],"affiliations":[{"raw_affiliation_string":"Shanghai Jiao Tong University, China","institution_ids":["https://openalex.org/I183067930"]}]},{"author_position":"last","author":{"id":"https://openalex.org/A5101471083","display_name":"Tianqi Chen","orcid":"https://orcid.org/0000-0002-2336-1875"},"institutions":[{"id":"https://openalex.org/I74973139","display_name":"Carnegie Mellon University","ror":"https://ror.org/05x2bcf33","country_code":"US","type":"education","lineage":["https://openalex.org/I74973139"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Tianqi Chen","raw_affiliation_strings":["Carnegie Mellon University, USA / OctoML, USA"],"affiliations":[{"raw_affiliation_string":"Carnegie Mellon University, USA / OctoML, USA","institution_ids":["https://openalex.org/I74973139"]}]}],"institutions":[],"countries_distinct_count":2,"institutions_distinct_count":11,"corresponding_author_ids":["https://openalex.org/A5015218854"],"corresponding_institution_ids":["https://openalex.org/I183067930"],"apc_list":null,"apc_paid":null,"fwci":19.9741,"has_fulltext":true,"cited_by_count":65,"citation_normalized_percentile":{"value":0.99739783,"is_in_top_1_percent":true,"is_in_top_10_percent":true},"cited_by_percentile_year":{"min":98,"max":100},"biblio":{"volume":null,"issue":null,"first_page":"804","last_page":"817"},"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.9988999962806702,"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.9988999962806702,"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.9824000000953674,"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/T12303","display_name":"Tensor decomposition and applications","score":0.9476000070571899,"subfield":{"id":"https://openalex.org/subfields/2605","display_name":"Computational Mathematics"},"field":{"id":"https://openalex.org/fields/26","display_name":"Mathematics"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/computer-science","display_name":"Computer science","score":0.846763014793396},{"id":"https://openalex.org/keywords/abstraction","display_name":"Abstraction","score":0.8417438268661499},{"id":"https://openalex.org/keywords/compiler","display_name":"Compiler","score":0.783414900302887},{"id":"https://openalex.org/keywords/computation","display_name":"Computation","score":0.6702129244804382},{"id":"https://openalex.org/keywords/tensor","display_name":"Tensor (intrinsic definition)","score":0.5961342453956604},{"id":"https://openalex.org/keywords/representation","display_name":"Representation (politics)","score":0.4936321973800659},{"id":"https://openalex.org/keywords/programming-language","display_name":"Programming language","score":0.48107171058654785},{"id":"https://openalex.org/keywords/set","display_name":"Set (abstract data type)","score":0.4721938371658325},{"id":"https://openalex.org/keywords/artificial-intelligence","display_name":"Artificial intelligence","score":0.4684588611125946},{"id":"https://openalex.org/keywords/class","display_name":"Class (philosophy)","score":0.42399200797080994},{"id":"https://openalex.org/keywords/theoretical-computer-science","display_name":"Theoretical computer science","score":0.4095270335674286},{"id":"https://openalex.org/keywords/parallel-computing","display_name":"Parallel computing","score":0.32124578952789307}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.846763014793396},{"id":"https://openalex.org/C124304363","wikidata":"https://www.wikidata.org/wiki/Q673661","display_name":"Abstraction","level":2,"score":0.8417438268661499},{"id":"https://openalex.org/C169590947","wikidata":"https://www.wikidata.org/wiki/Q47506","display_name":"Compiler","level":2,"score":0.783414900302887},{"id":"https://openalex.org/C45374587","wikidata":"https://www.wikidata.org/wiki/Q12525525","display_name":"Computation","level":2,"score":0.6702129244804382},{"id":"https://openalex.org/C155281189","wikidata":"https://www.wikidata.org/wiki/Q3518150","display_name":"Tensor (intrinsic definition)","level":2,"score":0.5961342453956604},{"id":"https://openalex.org/C2776359362","wikidata":"https://www.wikidata.org/wiki/Q2145286","display_name":"Representation (politics)","level":3,"score":0.4936321973800659},{"id":"https://openalex.org/C199360897","wikidata":"https://www.wikidata.org/wiki/Q9143","display_name":"Programming language","level":1,"score":0.48107171058654785},{"id":"https://openalex.org/C177264268","wikidata":"https://www.wikidata.org/wiki/Q1514741","display_name":"Set (abstract data type)","level":2,"score":0.4721938371658325},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.4684588611125946},{"id":"https://openalex.org/C2777212361","wikidata":"https://www.wikidata.org/wiki/Q5127848","display_name":"Class (philosophy)","level":2,"score":0.42399200797080994},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.4095270335674286},{"id":"https://openalex.org/C173608175","wikidata":"https://www.wikidata.org/wiki/Q232661","display_name":"Parallel computing","level":1,"score":0.32124578952789307},{"id":"https://openalex.org/C138885662","wikidata":"https://www.wikidata.org/wiki/Q5891","display_name":"Philosophy","level":0,"score":0.0},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.0},{"id":"https://openalex.org/C199539241","wikidata":"https://www.wikidata.org/wiki/Q7748","display_name":"Law","level":1,"score":0.0},{"id":"https://openalex.org/C202444582","wikidata":"https://www.wikidata.org/wiki/Q837863","display_name":"Pure mathematics","level":1,"score":0.0},{"id":"https://openalex.org/C17744445","wikidata":"https://www.wikidata.org/wiki/Q36442","display_name":"Political science","level":0,"score":0.0},{"id":"https://openalex.org/C94625758","wikidata":"https://www.wikidata.org/wiki/Q7163","display_name":"Politics","level":2,"score":0.0},{"id":"https://openalex.org/C111472728","wikidata":"https://www.wikidata.org/wiki/Q9471","display_name":"Epistemology","level":1,"score":0.0}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.1145/3575693.3576933","is_oa":true,"landing_page_url":"https://doi.org/10.1145/3575693.3576933","pdf_url":"https://dl.acm.org/doi/pdf/10.1145/3575693.3576933","source":null,"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the 28th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 2","raw_type":"proceedings-article"}],"best_oa_location":{"id":"doi:10.1145/3575693.3576933","is_oa":true,"landing_page_url":"https://doi.org/10.1145/3575693.3576933","pdf_url":"https://dl.acm.org/doi/pdf/10.1145/3575693.3576933","source":null,"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the 28th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 2","raw_type":"proceedings-article"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"pdf":true,"grobid_xml":true},"content_urls":{"pdf":"https://content.openalex.org/works/W4318541538.pdf","grobid_xml":"https://content.openalex.org/works/W4318541538.grobid-xml"},"referenced_works_count":20,"referenced_works":["https://openalex.org/W1486289306","https://openalex.org/W1991009705","https://openalex.org/W2137613189","https://openalex.org/W2194775991","https://openalex.org/W2295598076","https://openalex.org/W2590246587","https://openalex.org/W2606722458","https://openalex.org/W2949967139","https://openalex.org/W2954698171","https://openalex.org/W2961619211","https://openalex.org/W2999521721","https://openalex.org/W3012249773","https://openalex.org/W3090487264","https://openalex.org/W3098389819","https://openalex.org/W3108012228","https://openalex.org/W3122286897","https://openalex.org/W3123639931","https://openalex.org/W3177452048","https://openalex.org/W4251637954","https://openalex.org/W4281932226"],"related_works":["https://openalex.org/W1977285665","https://openalex.org/W2740990710","https://openalex.org/W2479014312","https://openalex.org/W2192862863","https://openalex.org/W1497385637","https://openalex.org/W1608659702","https://openalex.org/W1583465708","https://openalex.org/W2007728761","https://openalex.org/W2369288331","https://openalex.org/W1504651774"],"abstract_inverted_index":{"Deploying":[0],"deep":[1,95],"learning":[2,38,71,96],"models":[3,97],"on":[4,87],"various":[5],"devices":[6],"has":[7],"become":[8],"an":[9,84],"important":[10],"topic.":[11],"The":[12],"wave":[13],"of":[14,21,89],"hardware":[15,117],"specialization":[16],"brings":[17],"a":[18,50],"diverse":[19],"set":[20],"acceleration":[22,31],"primitives":[23,114],"for":[24,53,98,115],"multi-dimensional":[25],"ten-":[26],"sor":[27],"computations.":[28],"These":[29],"new":[30],"primitives,":[32],"along":[33],"with":[34,56],"the":[35,63,78,111],"emerging":[36],"machine":[37,70],"models,":[39],"bring":[40,74],"tremendous":[41],"engineering":[42],"challenges.":[43],"In":[44],"this":[45],"paper,":[46],"we":[47,82],"present":[48],"TensorIR,":[49],"compiler":[51],"abstraction":[52,91],"optimizing":[54],"programs":[55],"these":[57],"tensor":[58,75,100,112],"computation":[59,76,101,113],"primitives.":[60,102],"TensorIR":[61,107],"generalizes":[62],"loop":[64],"nest":[65],"representation":[66],"used":[67],"in":[68],"existing":[69],"compilers":[72],"to":[73,92,125],"as":[77],"first-class":[79],"citizen.":[80],"Finally,":[81],"build":[83],"end-to-end":[85],"framework":[86],"top":[88],"our":[90],"automatically":[93,109],"optimize":[94],"given":[99,116],"Experimental":[103],"results":[104],"show":[105],"that":[106,122],"compilation":[108],"uses":[110],"backends":[118],"and":[119],"delivers":[120],"performance":[121],"is":[123],"competitive":[124],"state-of-art":[126],"hand-optimized":[127],"systems":[128],"across":[129],"platforms.":[130]},"counts_by_year":[{"year":2026,"cited_by_count":10},{"year":2025,"cited_by_count":26},{"year":2024,"cited_by_count":22},{"year":2023,"cited_by_count":7}],"updated_date":"2026-03-14T08:43:22.919905","created_date":"2025-10-10T00:00:00"}