[Nvidia][TritonGPU] Add phases to RemoveLayoutConversions

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: zeroherolin

CMakeLists.txt

@@ -35,6 +35,26 @@ endif()
 if(NOT FLAGTREE_BACKEND)
   add_definitions(-D__NVIDIA__)
   add_definitions(-D__AMD__)
+  option(TRITON_RLC_ENABLE_COST_BASED_RESOLUTION
+         "RLC: cost-based layout conflict resolution" ON)
+  option(TRITON_RLC_ENABLE_BACKWARD_PROPAGATION
+         "RLC: backward layout propagation from writeback converts" ON)
+  option(TRITON_RLC_ENABLE_SMALL_COMPONENT_SOLVING
+         "RLC: small closed-component solving" ON)
+  option(TRITON_RLC_ENABLE_STORE_LAYOUT_REMATERIALIZATION
+         "RLC: store-layout rematerialization for MMA writebacks" ON)
+  if(TRITON_RLC_ENABLE_COST_BASED_RESOLUTION)
+    add_definitions(-DTRITON_RLC_ENABLE_COST_BASED_RESOLUTION)
+  endif()
+  if(TRITON_RLC_ENABLE_BACKWARD_PROPAGATION)
+    add_definitions(-DTRITON_RLC_ENABLE_BACKWARD_PROPAGATION)
+  endif()
+  if(TRITON_RLC_ENABLE_SMALL_COMPONENT_SOLVING)
+    add_definitions(-DTRITON_RLC_ENABLE_SMALL_COMPONENT_SOLVING)
+  endif()
+  if(TRITON_RLC_ENABLE_STORE_LAYOUT_REMATERIALIZATION)
+    add_definitions(-DTRITON_RLC_ENABLE_STORE_LAYOUT_REMATERIALIZATION)
+  endif()
 elseif(FLAGTREE_BACKEND STREQUAL "iluvatar")
   add_definitions(-D__ILUVATAR__)
   set(FLAGTREE_TLE OFF)