Use correct TileScheduleType

Signed-off-by: Lukas Sommer <[email protected]>
codeplaysoftware · Dec 13, 2024 · 19524ee · 19524ee
1 parent 341999d
commit 19524ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)