[all-commits] [llvm/llvm-project] e534ce: [mlir] Add a shape function library op

Jacques Pienaar via All-commits all-commits at lists.llvm.org
Sun Nov 29 11:15:58 PST 2020


  Branch: refs/heads/master
  Home:   https://github.com/llvm/llvm-project
  Commit: e534cee26ae3626ced20438ea82e11291cc768e8
      https://github.com/llvm/llvm-project/commit/e534cee26ae3626ced20438ea82e11291cc768e8
  Author: Jacques Pienaar <jpienaar at google.com>
  Date:   2020-11-29 (Sun, 29 Nov 2020)

  Changed paths:
    M mlir/include/mlir/Dialect/Shape/IR/Shape.h
    M mlir/include/mlir/Dialect/Shape/IR/ShapeOps.td
    M mlir/lib/Dialect/Shape/IR/Shape.cpp
    A mlir/test/Analysis/test-shape-fn-report.mlir
    M mlir/test/lib/Dialect/CMakeLists.txt
    A mlir/test/lib/Dialect/Shape/CMakeLists.txt
    A mlir/test/lib/Dialect/Shape/TestShapeFunctions.cpp
    M mlir/test/lib/Dialect/Test/TestOps.td
    M mlir/tools/mlir-opt/CMakeLists.txt
    M mlir/tools/mlir-opt/mlir-opt.cpp

  Log Message:
  -----------
  [mlir] Add a shape function library op

Op with mapping from ops to corresponding shape functions for those op
in the library and mechanism to associate shape functions to functions.
The mapping of operand to shape function is kept separate from the shape
functions themselves as the operation is associated to the shape
function and not vice versa, and one could have a common library of
shape functions that can be used in different contexts.

Use fully qualified names and require a name for shape fn lib ops for
now and an explicit print/parse (based around the generated one & GPU
module op ones).

This commit reverts d9da4c3e73720badfcac5c0dc63c0285bb690770. Fixes
missing headers (don't know how that was working locally).

Differential Revision: https://reviews.llvm.org/D91672




More information about the All-commits mailing list