From 7192f4ab230bb721fa8d4d3df33886dbe86cdc59 Mon Sep 17 00:00:00 2001
From: Tri Dao <tridao@users.noreply.github.com>
Date: Thu, 8 Aug 2024 11:00:24 -0700
Subject: [PATCH] Add CLayout_64x208 (#1680)

Without this I get compilation error when the extended shapes are enabled
---
 include/cute/atom/mma_traits_sm90_gmma.hpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/include/cute/atom/mma_traits_sm90_gmma.hpp b/include/cute/atom/mma_traits_sm90_gmma.hpp
index 3a4fdfa1..e59bbeef 100644
--- a/include/cute/atom/mma_traits_sm90_gmma.hpp
+++ b/include/cute/atom/mma_traits_sm90_gmma.hpp
@@ -422,6 +422,9 @@ using CLayout_64x176  = Layout<Shape <Shape <  _4,_8, _4>,Shape < _2,_2, Int<22>
 using CLayout_64x192 = Layout<Shape <Shape <  _4,_8, _4>,Shape < _2,_2, _24>>,
                               Stride<Stride<_128,_1,_16>,Stride<_64,_8,_512>>>;
 
+using CLayout_64x208  = Layout<Shape <Shape <  _4,_8, _4>,Shape < _2,_2, Int<26>>>,
+                              Stride<Stride<_128,_1,_16>,Stride<_64,_8,_512>>>;
+
 using CLayout_64x224  = Layout<Shape <Shape <  _4,_8, _4>,Shape < _2,_2, Int<28>>>,
                               Stride<Stride<_128,_1,_16>,Stride<_64,_8,_512>>>;