feat: implement TileLang-based DSA split-K indexer loss, add regression tests, and clamp warmup_steps in ScheduleEditor

Author: apstenku123

backend.log

@@ -810,6 +810,12 @@ def wrap_tilelang_metal_kernel(
                 "output_buffer_names contains buffers missing from the Metal "
                 f"signature: {missing_outputs!r}; parsed={buffer_names!r}"
             )
+        aliased_names = set(input_sources) & set(output_sources)
+        if aliased_names:
+            raise MLXRuntimeError(
+                f"Input and output buffer names must be mutually disjoint (aliasing is not supported): "
+                f"aliased={sorted(list(aliased_names))}"
+            )
         explicit_sources = input_sources + output_sources
         if len(set(explicit_sources)) != len(explicit_sources):
             raise MLXRuntimeError(
@@ -833,6 +839,14 @@ def wrap_tilelang_metal_kernel(
             f"{buffer_names!r}"
         )
 
+    # Ensure no input/output aliasing overlap
+    aliased_names = set(input_sources) & set(output_sources)
+    if aliased_names:
+        raise MLXRuntimeError(
+            f"Input and output buffer names must be mutually disjoint (aliasing is not supported): "
+            f"aliased={sorted(list(aliased_names))}"
+        )
+
     input_names = tuple(f"inp{i}" for i in range(len(input_sources)))
     output_names = tuple(f"out{i}" for i in range(len(output_sources)))
     rename: dict[str, str] = {}