LEARNED GRAPH OPTIMIZATIONS FOR COMPILERS
    8.
    发明公开

    公开(公告)号:US20230176840A1

    公开(公告)日:2023-06-08

    申请号:US17921933

    申请日:2021-06-07

    Applicant: Google LLC

    CPC classification number: G06F8/443

    Abstract: Methods, systems, and apparatus, including computer programs encoded on computer storage media, for compiler optimizations using a compiler optimization network. One of the methods includes receiving an input program, wherein the input program defines a graph of operation modules, wherein each node in the graph is a respective operation module, and each edge between nodes in the graph represents one operation module receiving the output generated by another operation module. The input program is processed by a compiler optimization network comprising a graph-embedding network that is configured to encode operation features and operation dependencies of the operation modules of the input program into a graph embedding representation and a policy network that is configured to generate an optimization action for each of one or more nodes encoded in the graph embedding representation. The compiler optimization network generates an output optimization plan comprising one or more optimization actions for the input program.

Patent Agency Ranking