From 6c5a897c447b5294e557ce3f22b08e5cb333c443 Mon Sep 17 00:00:00 2001
From: Cullen Rhodes <cullen.rhodes@arm.com>
Date: Wed, 3 Nov 2021 11:49:58 +0000
Subject: [PATCH] [mlir][nvvm] NFC: Fix unused template arg tablegen warning

Identified in D109359.
---
 mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td b/mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td
index b7aa501..db76cc1 100644
--- a/mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td
+++ b/mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td
@@ -195,12 +195,12 @@ class MMA_SIGNATURE<WMMA_REGS A, WMMA_REGS B, WMMA_REGS C, WMMA_REGS D> {
 }
 
 /// Generate enum value of the wmma.mma intrinsic.
-class WMMA_NAME<string ALayout, string BLayout, WMMA_REGS A, WMMA_REGS B,
-  WMMA_REGS C, WMMA_REGS D> {
+class WMMA_NAME<string Op, string ALayout, string BLayout, WMMA_REGS A,
+  WMMA_REGS B, WMMA_REGS C, WMMA_REGS D> {
   string signature = MMA_SIGNATURE<A, B, C, D>.ret;
   string id =   "llvm::Intrinsic::nvvm_wmma"
                 # "_" # A.geom
-                # "_mma"
+                # "_" # Op
                 # "_" # ALayout
                 # "_" # BLayout
                 # signature;
@@ -342,7 +342,7 @@ class MMA_MMA_INTR<string opName> {
   "    && \"" # op[0].ptx_elt_type # "\" == eltypeA && \""
    # op[3].ptx_elt_type # "\" == eltypeB)"
   "  return " #
-       WMMA_NAME<layoutA, layoutB, op[0], op[1], op[2], op[3]>.id # ";")));
+       WMMA_NAME<opName, layoutA, layoutB, op[0], op[1], op[2], op[3]>.id # ";")));
   list<string> f = !foldl([""],
                      !foldl([[""]], cond0, acc, el, !listconcat(acc, el)),
                           acc1, el1, !listconcat(acc1, el1));
-- 
2.7.4