[all-commits] [llvm/llvm-project] e552fa: [MLIR][GPU] Add CUDA Tensor core WMMA test

[Navdeep Kumar via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: e552fa28da286f20f963d51dd05bd3ec278553b7
      https://github.com/llvm/llvm-project/commit/e552fa28da286f20f963d51dd05bd3ec278553b7
  Author: Navdeep Kumar <navdeep.navdeep37 at gmail.com>
  Date:   2021-05-22 (Sat, 22 May 2021)

  Changed paths:
    M mlir/test/CMakeLists.txt
    A mlir/test/Integration/GPU/CUDA/TensorCore/lit.local.cfg
    A mlir/test/Integration/GPU/CUDA/TensorCore/wmma-matmul-f16.mlir
    A mlir/test/Integration/GPU/CUDA/TensorCore/wmma-matmul-f32.mlir
    M mlir/test/lit.site.cfg.py.in

  Log Message:
  -----------
  [MLIR][GPU] Add CUDA Tensor core WMMA test

Add a test case to test the complete execution of WMMA ops on a Nvidia
GPU with tensor cores. These tests are enabled under
MLIR_RUN_CUDA_TENSOR_CORE_TESTS.

Reviewed By: bondhugula

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