[all-commits] [llvm/llvm-project] 31f888: [mlir][linalg][python] Add attribute support to th...

Tobias Gysi via All-commits all-commits at lists.llvm.org
Thu Jun 24 02:41:47 PDT 2021


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 31f888ea9af452ae312c270e569d9fbe23c57c9f
      https://github.com/llvm/llvm-project/commit/31f888ea9af452ae312c270e569d9fbe23c57c9f
  Author: Tobias Gysi <gysit at google.com>
  Date:   2021-06-24 (Thu, 24 Jun 2021)

  Changed paths:
    M mlir/include/mlir-c/AffineMap.h
    M mlir/lib/Bindings/Python/IRAffine.cpp
    M mlir/lib/CAPI/IR/AffineMap.cpp
    M mlir/python/mlir/dialects/linalg/opdsl/lang/comprehension.py
    M mlir/python/mlir/dialects/linalg/opdsl/lang/config.py
    M mlir/python/mlir/dialects/linalg/opdsl/lang/dsl.py
    M mlir/python/mlir/dialects/linalg/opdsl/lang/emitter.py
    M mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py
    M mlir/test/python/dialects/linalg/opdsl/arguments.py
    M mlir/test/python/dialects/linalg/opdsl/emit_structured_generic.py
    M mlir/test/python/dialects/linalg/opdsl/shape_maps_iteration.py
    M mlir/test/python/dialects/linalg/opsrun.py
    M mlir/test/python/ir/affine_map.py

  Log Message:
  -----------
  [mlir][linalg][python] Add attribute support to the OpDSL.

Extend the OpDSL with index attributes. After tensors and scalars, index attributes are the third operand type. An index attribute represents a compile-time constant that is limited to index expressions. A use cases are the strides and dilations defined by convolution and pooling operations.

The patch only updates the OpDSL. The C++ yaml codegen is updated by a followup patch.

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




More information about the All-commits mailing list