fix: COL_MAJOR
This commit is contained in:
@@ -44,7 +44,7 @@ def main():
|
||||
|
||||
# V FMHA layout: use explicit constants
|
||||
# pv_n_tile=256, s_k=128
|
||||
v_major = LayoutEnum.COLUMN_MAJOR # layout (256, 128, 1) stride (1, 256, 32768) = col-major
|
||||
v_major = LayoutEnum.COL_MAJOR # layout (256, 128, 1) stride (1, 256, 32768) = col-major
|
||||
|
||||
c_layout = LayoutEnum.from_tensor(mC)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user