From 4b11851cede64a3dcf589ded266cafad8236d787 Mon Sep 17 00:00:00 2001
From: root <root@DESKTOP-RFK1F29.>
Date: Wed, 31 Jul 2024 21:22:25 +0800
Subject: [PATCH] fix GemmFpAIntB MMa::IteratorB::Layout

---
 .../include/cutlass_extensions/gemm/kernel/fpA_intB_gemm.h      | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cpp/tensorrt_llm/cutlass_extensions/include/cutlass_extensions/gemm/kernel/fpA_intB_gemm.h b/cpp/tensorrt_llm/cutlass_extensions/include/cutlass_extensions/gemm/kernel/fpA_intB_gemm.h
index 63469f225..569517c63 100644
--- a/cpp/tensorrt_llm/cutlass_extensions/include/cutlass_extensions/gemm/kernel/fpA_intB_gemm.h
+++ b/cpp/tensorrt_llm/cutlass_extensions/include/cutlass_extensions/gemm/kernel/fpA_intB_gemm.h
@@ -80,7 +80,7 @@ struct GemmFpAIntB
     using ElementA = typename Mma::IteratorA::Element;
     using LayoutA = typename Mma::IteratorA::Layout;
     using ElementB = typename Mma::IteratorB::Element;
-    using LayoutB = typename Mma::IteratorB::Element;
+    using LayoutB = typename Mma::IteratorB::Layout;
     using ElementC = typename Epilogue::OutputTileIterator::Element;
     using LayoutC = typename Mma::LayoutC;
     using ElementScale = ElementC;