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

Add missing schedules argument in SM90 fp16 op generation (#920)

Browse Source
This commit is contained in:
Adnan Akhundov 2023-04-26 22:44:49 +02:00 committed by GitHub
parent 180c5629bf
commit df02482f1d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/library/scripts/generator.py
View File

@ -4160,7 +4160,7 @@ def GenerateSM90_TensorOp_16b_WGMMA_gemm(manifest, cuda_version):
elif data_type_mixed["c_type"] in [DataType.f16, DataType.bf16]: elif data_type_mixed["c_type"] in [DataType.f16, DataType.bf16]:
layout[2][1] = 8 layout[2][1] = 8
CreateGemmUniversal3xOperator(manifest, layouts, tile_descriptions, data_type_mixed) CreateGemmUniversal3xOperator(manifest, layouts, tile_descriptions, data_type_mixed, schedules)
# persistent kernels with TMA epilogues # persistent kernels with TMA epilogues
if data_type_mixed["c_type"] in [DataType.f16, DataType.bf16] and CudaToolkitVersionSatisfies(cuda_version, 12, 1): if data_type_mixed["c_type"] in [DataType.f16, DataType.bf16] and CudaToolkitVersionSatisfies(cuda_version, 12, 1):
CreateGemmUniversal3xOperator(manifest, layouts, tile_descriptions, data_type_mixed, CreateGemmUniversal3xOperator(manifest, layouts, tile_descriptions, data_type_mixed,