Apply code-format changes

Author: flagtree-bot

python/test/tle/unit/test_tle.py

@@ -24,6 +24,7 @@
 from triton.experimental.tle.language.gpu.semantic import TLESemanticError, TLESemantic
 import triton._C.libtriton as libtriton
 
+
 @triton.jit
 def _encoding_frontend_kernel(layout: tl.constexpr):
     x = tl.arange(0, 128)
@@ -42,8 +43,10 @@ def _dot_encoding_frontend_kernel(
     acc = tle.gpu.set_layout(tl.zeros((32, 8), tl.float32), mma_layout)
     result = tl.dot(lhs, rhs, acc=acc, out_dtype=tl.float32)  # noqa: F841
 
+
 _HAS_TLE_EXPLICIT_LAYOUT = hasattr(libtriton.ir.builder, "ensure_ttg_layout_attrs")
 
+
 def _cuda_backend_available():
     from triton.compiler.compiler import get_backend
     try:
@@ -52,6 +55,7 @@ def _cuda_backend_available():
     except Exception:
         return False
 
+
 class TestLayoutEncoding:
     """Test layout encoding"""
 
@@ -70,7 +74,7 @@ def test_swizzled_shared_layout_permute(self):
         # Original order for 3D rank is [2, 1, 0]
         # Permuting with [1, 0, 2] gives: order[1], order[0], order[2] = [1, 2, 0]
         assert permuted.order == (1, 2, 0)
-    
+
     @pytest.mark.skipif(not _HAS_TLE_EXPLICIT_LAYOUT, reason="requires __TLE__ build")
     @pytest.mark.skipif(not _cuda_backend_available(), reason="requires cuda backend")
     def test_explicit_distributed_encoding_frontend(self):
@@ -87,7 +91,7 @@ def test_explicit_distributed_encoding_frontend(self):
         assert "tle.gpu.set_layout" in ir
         assert "ttg.convert_layout" not in ir
         assert "#ttg.slice<{dim = 0, parent = #blocked}>" in ir
-    
+
     @pytest.mark.skipif(not _HAS_TLE_EXPLICIT_LAYOUT, reason="requires __TLE__ build")
     @pytest.mark.skipif(not _cuda_backend_available(), reason="requires cuda backend")
     def test_explicit_dot_operand_encoding_frontend(self):

python/triton/experimental/tle/language/gpu/core.py

@@ -64,12 +64,10 @@ def set_layout(value, layout, _semantic=None):
     if not value.type.is_block():
         raise ValueError("tle.gpu.set_layout only supports block tensors")
     if not hasattr(_semantic.builder, "ensure_ttg_layout_attrs"):
-        raise RuntimeError(
-            "tle.gpu.set_layout requires a Triton build with __TLE__ explicit "
-            "layout support (ir.builder.ensure_ttg_layout_attrs is missing). "
-            "This backend likely uses its own TLE variant "
-            "(e.g. __ILUVATAR_TLE__/__MCTLE__) that does not implement this op."
-        )
+        raise RuntimeError("tle.gpu.set_layout requires a Triton build with __TLE__ explicit "
+                           "layout support (ir.builder.ensure_ttg_layout_attrs is missing). "
+                           "This backend likely uses its own TLE variant "
+                           "(e.g. __ILUVATAR_TLE__/__MCTLE__) that does not implement this op.")
     options = _semantic.builder.options
     _semantic.builder.ensure_ttg_layout_attrs(
         int(getattr(options, "num_warps")),