[mlir][linalg] Improve tensor.extract vectorisation when dynamic shapes are present

llvm / llvm-project

The LLVM Project is a collection of modular and reusable compiler and toolchain technologies.

http://llvm.org

Other

27.84k stars 11.47k forks source link

[mlir][linalg] Improve tensor.extract vectorisation when dynamic shapes are present #63918

Open banach-space opened 1 year ago

banach-space commented 1 year ago

As per Diego's comment, make sure that the tensor.extract in lit test is vectorised as a broadcast if only dimension 2 is vectorised: https://reviews.llvm.org/D155373

llvmbot commented 1 year ago

@llvm/issue-subscribers-mlir-linalg