[all-commits] [llvm/llvm-project] 670eee: [mlir][VectorToGPU] Fix support for i4, col-major ...

[Chris via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 670eee08cecfcfe170fb0e7daa88df8c2a150dbe
      https://github.com/llvm/llvm-project/commit/670eee08cecfcfe170fb0e7daa88df8c2a150dbe
  Author: Christopher Bate <cbate at nvidia.com>
  Date:   2022-06-30 (Thu, 30 Jun 2022)

  Changed paths:
    M mlir/lib/Conversion/VectorToGPU/NvGpuSupport.cpp
    M mlir/lib/Conversion/VectorToGPU/VectorToGPU.cpp
    M mlir/test/Conversion/VectorToGPU/vector-to-mma-ops-mma-sync.mlir

  Log Message:
  -----------
  [mlir][VectorToGPU] Fix support for i4, col-major operand support

For the conversion to nvgpu `mma.sync` and `ldmatrix` pathways, the code
was missing support for the `i4` data type. While fixing this, another
bug was discoverd that caused the number of ldmatrix tiles calculated for
certain operand types and configurations to be incorrect. This change
fixes both issues and adds additional tests.

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