[all-commits] [llvm/llvm-project] a9efcb: [MLIR] Add continuous tiling to transform dialect ...

[muneebkhan85 via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: a9efcbf490d9b8f46ec37062ca8653b4068000e5
      https://github.com/llvm/llvm-project/commit/a9efcbf490d9b8f46ec37062ca8653b4068000e5
  Author: muneebkhan85 <150162960+muneebkhan85 at users.noreply.github.com>
  Date:   2024-06-21 (Fri, 21 Jun 2024)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.td
    M mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
    M mlir/include/mlir/Dialect/Transform/IR/TransformOps.td
    M mlir/lib/Dialect/Linalg/TransformOps/LinalgTransformOps.cpp
    M mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp
    M mlir/lib/Dialect/Transform/IR/TransformOps.cpp
    M mlir/python/mlir/dialects/transform/structured.py
    A mlir/test/Dialect/Linalg/continuous-tiling-full.mlir
    A mlir/test/Dialect/Linalg/continuous-tiling-multiway-split.mlir
    M mlir/test/Dialect/Linalg/transform-op-split.mlir
    M mlir/test/python/dialects/transform_structured_ext.py

  Log Message:
  -----------
  [MLIR] Add continuous tiling to transform dialect (#82792)

This patch enables continuous tiling of a target structured op using
diminishing tile sizes. In cases where the tensor dimensions are not
exactly divisible by the tile size, we are left with leftover tensor
chunks that are irregularly tiled. This approach enables tiling of the
leftover chunk with a smaller tile size and repeats this process
recursively using exponentially diminishing tile sizes. This eventually
generates a chain of loops that apply tiling using diminishing tile
sizes.

Adds `continuous_tile_sizes` op to the transform dialect. This op, when
given a tile size and a dimension, computes a series of diminishing tile
sizes that can be used to tile the target along the given dimension.
Additionally, this op also generates a series of chunk sizes that the
corresponding tile sizes should be applied to along the given dimension.

Adds `multiway` attribute to `transform.structured.split` that enables
multiway splitting of a single target op along the given dimension, as
specified in a list enumerating the chunk sizes.



To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications