squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add CLayout_64x208 (#1680)

Browse Source 
Without this I get compilation error when the extended shapes are enabled
This commit is contained in:
Tri Dao 2024-08-08 11:00:24 -07:00 committed by GitHub
parent 2049c6c5a2
commit 7192f4ab23
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
include/cute/atom/mma_traits_sm90_gmma.hpp
View File

@ -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>>>;