Support Span to construct from a single element similar to ArrayRef (pytorch#11778)

This PR was created by the merge bot to help merge the original PR into
the main branch.
ghstack PR number: pytorch#11757 by
@cccclai
^ Please use this as the source of truth for the PR details, comments,
and reviews
ghstack PR base:
https://github.com/pytorch/executorch/tree/gh/cccclai/28/base
ghstack PR head:
https://github.com/pytorch/executorch/tree/gh/cccclai/28/head
Merge bot PR base: https://github.com/pytorch/executorch/tree/main
Merge bot PR head:
https://github.com/pytorch/executorch/tree/gh/cccclai/28/orig
@diff-train-skip-merge

Co-authored-by: Chen Lai <[email protected]>

Author: pytorchbot

runtime/core/span.h

@@ -55,6 +55,10 @@ class Span final {
   template <size_t N>
   /* implicit */ constexpr Span(T (&Arr)[N]) : data_(Arr), length_(N) {}
 
+  /// Construct a Span from a single element reference.
+  /* implicit */ constexpr Span(T& single_element)
+      : data_(&single_element), length_(1) {}
+
   /// @returns a pointer to the start of the underlying element buffer.
   iterator begin() const noexcept {
     return data_;

runtime/core/test/span_test.cpp

@@ -61,3 +61,19 @@ TEST(SpanTest, TriviallyCopyable) {
   EXPECT_EQ(span.size(), span_copy.size());
   EXPECT_TRUE(std::is_trivially_copyable<Span<int64_t>>::value);
 }
+
+TEST(SpanTest, SingleElementConstructor) {
+  int64_t single_value = 42;
+  Span<int64_t> span = single_value;
+
+  EXPECT_EQ(span.size(), 1);
+  EXPECT_EQ(span.data(), &single_value);
+  EXPECT_EQ(span[0], 42);
+  EXPECT_EQ(*span.begin(), 42);
+  EXPECT_EQ(span.end(), span.begin() + 1);
+
+  // Test that modifying through span affects original value
+  span[0] = 100;
+  EXPECT_EQ(single_value, 100);
+  EXPECT_EQ(span[0], 100);
+}