From 19524ee91fcb230570435aefa95b58aec8ea43ad Mon Sep 17 00:00:00 2001 From: Lukas Sommer Date: Thu, 14 Nov 2024 14:02:59 +0000 Subject: [PATCH] Use correct TileScheduleType Signed-off-by: Lukas Sommer --- python/cutlass_library/generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/cutlass_library/generator.py b/python/cutlass_library/generator.py index efe90ab57..ac37d2d16 100644 --- a/python/cutlass_library/generator.py +++ b/python/cutlass_library/generator.py @@ -6936,7 +6936,7 @@ def GeneratePVC_TensorOp_16b_gemm(manifest, cuda_version): schedules = [[KernelScheduleType.ScheduleAuto, EpilogueScheduleType.ScheduleAuto]] - CreateGemmUniversal3xOperator(manifest, layouts, tile_descriptions, data_type, schedules, tile_schedulers=[TileSchedulerType.Default]) + CreateGemmUniversal3xOperator(manifest, layouts, tile_descriptions, data_type, schedules, tile_schedulers=[TileSchedulerType.Persistent]) def GeneratePVC(manifest, cuda_version): GeneratePVC_TensorOp_16b_gemm(manifest, cuda_version)