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)