[FRONTEND] .item() as syntactic sugar for .reshape([]) (#6873)

suggested by @lezcano

Author: apgoucher

python/test/unit/language/test_core.py

@@ -7528,7 +7528,7 @@ def unsplat_kernel(x, explicit: tl.constexpr):
         condition = tl.load(x + tl.arange(0, 1)) > 42
 
         if explicit:
-            condition = condition.reshape([])
+            condition = condition.item()
 
         if condition:
             tl.store(x, 42)

python/triton/compiler/code_generator.py

@@ -773,7 +773,7 @@ def visit_If(self, node):
                 raise self._unsupported(node, "Boolean value of Tensor with more than one value is ambiguous")
             if cond.type.is_block():
                 warnings.warn(
-                    "If conditional called with multidimensional Tensor instead of scalar; please use \"if (%s).reshape([])\" instead"
+                    "If conditional called with multidimensional Tensor instead of scalar; please use \"if (%s).item()\" instead"
                     % ast.unparse(node.test))
                 cond = language.core._unsplat(cond, _builder=self.builder, _generator=self)
             cond = cond.to(language.int1, _builder=self.builder)

python/triton/language/core.py

@@ -1792,6 +1792,15 @@ def view(input, *shape, _builder=None):
     return semantic.reshape(input, shape, can_reorder=True, builder=_builder)
 
 
+@_tensor_member_fn
+@builtin
+def item(input, _builder=None, _generator=None):
+    """
+    Converts a single-element tensor into a scalar.
+    """
+    return _unsplat(input, _builder=_builder, _generator=_generator)
+
+
 @_tensor_member_fn
 @builtin
 def reshape(input, *shape, can_reorder=False, _builder=None, _generator=None):