Skip to content

Update MutableSpan and OutputSpan #85481

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 6 commits into from
Nov 14, 2025
Merged

Update MutableSpan and OutputSpan #85481

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
c5fd69d
Update bounds check tests to use stdlib's MutableSpan and OutputSpan
meg-gupta Jul 17, 2025
2365d12
Update check lines after using standard library's MutableSpan and Out…
meg-gupta Jul 17, 2025
40a2ecb
Update subscript and count for MutableSpan and OutputSpan with @_sema…
meg-gupta Jul 17, 2025
b8376fd
Add some read only tests for MutableSpan
meg-gupta Jul 18, 2025
f0839be
Update MutableSpan bounds check tests
meg-gupta Nov 13, 2025
c93163c
Add _assumeNonNegative to OutputSpan's count as well
meg-gupta Nov 13, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 10 additions & 3 deletions stdlib/public/core/Span/MutableSpan.swift
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -236,6 +236,7 @@ extension MutableSpan where Element: ~Copyable {
extension MutableSpan where Element: ~Copyable {

@_alwaysEmitIntoClient
@_semantics("fixed_storage.get_count")
public var count: Int { _assumeNonNegative(_count) }

@_alwaysEmitIntoClient
Expand Down Expand Up @@ -279,7 +280,13 @@ extension MutableSpan where Element: BitwiseCopyable {
@available(SwiftCompatibilitySpan 5.0, *)
@_originallyDefinedIn(module: "Swift;CompatibilitySpan", SwiftCompatibilitySpan 6.2)
extension MutableSpan where Element: ~Copyable {

// SILOptimizer looks for fixed_storage.check_index semantics for bounds check optimizations.
@_semantics("fixed_storage.check_index")
@inline(__always)
@_alwaysEmitIntoClient
internal func _checkIndex(_ position: Index) {
_precondition(indices.contains(position), "index out of bounds")
}
/// Accesses the element at the specified position in the `MutableSpan`.
///
/// - Parameter position: The offset of the element to access. `position`
Expand All @@ -289,12 +296,12 @@ extension MutableSpan where Element: ~Copyable {
@_alwaysEmitIntoClient
public subscript(_ position: Index) -> Element {
unsafeAddress {
_precondition(indices.contains(position), "index out of bounds")
_checkIndex(position)
return unsafe UnsafePointer(_unsafeAddressOfElement(unchecked: position))
}
@lifetime(self: copy self)
unsafeMutableAddress {
_precondition(indices.contains(position), "index out of bounds")
_checkIndex(position)
return unsafe _unsafeAddressOfElement(unchecked: position)
}
}
Expand Down
17 changes: 13 additions & 4 deletions stdlib/public/core/Span/OutputSpan.swift
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,8 @@ extension OutputSpan where Element: ~Copyable {
extension OutputSpan where Element: ~Copyable {
/// The number of initialized elements in this span.
@_alwaysEmitIntoClient
public var count: Int { _count }
@_semantics("fixed_storage.get_count")
public var count: Int { _assumeNonNegative(_count) }

/// The number of additional elements that can be added to this span.
@_alwaysEmitIntoClient
Expand Down Expand Up @@ -190,7 +191,15 @@ extension OutputSpan where Element: ~Copyable {
/// The range of initialized positions for this `OutputSpan`.
@_alwaysEmitIntoClient
public var indices: Range<Index> {
unsafe Range(_uncheckedBounds: (0, _count))
unsafe Range(_uncheckedBounds: (0, count))
}

// SILOptimizer looks for fixed_storage.check_index semantics for bounds check optimizations.
@_semantics("fixed_storage.check_index")
@inline(__always)
@_alwaysEmitIntoClient
internal func _checkIndex(_ index: Index) {
_precondition(indices.contains(index), "index out of bounds")
}

/// Accesses the element at the specified position.
Expand All @@ -201,12 +210,12 @@ extension OutputSpan where Element: ~Copyable {
@_alwaysEmitIntoClient
public subscript(_ index: Index) -> Element {
unsafeAddress {
_precondition(indices.contains(index), "index out of bounds")
_checkIndex(index)
return unsafe UnsafePointer(_unsafeAddressOfElement(unchecked: index))
}
@lifetime(self: copy self)
unsafeMutableAddress {
_precondition(indices.contains(index), "index out of bounds")
_checkIndex(index)
return unsafe _unsafeAddressOfElement(unchecked: index)
}
}
Expand Down
1 change: 1 addition & 0 deletions stdlib/public/core/Span/Span.swift
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -414,6 +414,7 @@ extension Span where Element: ~Copyable {
@available(SwiftCompatibilitySpan 5.0, *)
@_originallyDefinedIn(module: "Swift;CompatibilitySpan", SwiftCompatibilitySpan 6.2)
extension Span where Element: ~Copyable {
// SILOptimizer looks for fixed_storage.check_index semantics for bounds check optimizations.
@_semantics("fixed_storage.check_index")
@inline(__always)
@_alwaysEmitIntoClient
Expand Down
Loading