METHOD AND APPARATUS FOR GENERATING TILING STRATEGY FOR TENSOR COMPUTATION

    公开(公告)号:US20250124106A1

    公开(公告)日:2025-04-17

    申请号:US19002417

    申请日:2024-12-26

    Abstract: A method for generating a tiling strategy for tensor computation is provided, including obtaining information about a plurality of tensor operations corresponding to tensor computation, where information about each tensor operation includes a tensor computation dimension corresponding to the tensor operation, a data type of an element corresponding to the tensor computation dimension, and a priority of the tensor computation dimension; determining a correspondence between the plurality of tensor operations and a plurality of hardware units; obtaining, the data type of the element corresponding to the tensor computation dimension, and the priority of the tensor computation dimension; and obtaining the tiling strategy for the tensor computation.

Patent Agency Ranking