From 8e85580859f0a92bf172c10dedeb4fdf4b59d703 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tianqi=20Zhang=20=28=E5=BC=A0=E5=A4=A9=E5=90=AF=29?= Date: Thu, 20 Jul 2023 02:26:01 +0800 Subject: [PATCH] fix layout bug (#1006) --- python/cutlass/library_defaults.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/cutlass/library_defaults.py b/python/cutlass/library_defaults.py index f72ca394..997d5d35 100644 --- a/python/cutlass/library_defaults.py +++ b/python/cutlass/library_defaults.py @@ -128,7 +128,7 @@ class KernelsForDataType: :rtype: int """ # Determine the leading dimension of the shape - if layout == cutlass.LayoutType.RowMajor: + if layout == cutlass.LayoutType.ColumnMajor: ld = shape[0] elif layout == cutlass.LayoutType.RowMajor: ld = shape[1]