Skip to content

Commit a6e2961

Browse files
authored
Mark native Swift extension public (#11087)
1 parent f252d19 commit a6e2961

File tree

2 files changed

+16
-16
lines changed

2 files changed

+16
-16
lines changed

extension/apple/ExecuTorch/Exported/ExecuTorch+Tensor.swift

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -11,40 +11,40 @@
1111
/// A protocol that types conform to in order to be used as tensor element types.
1212
/// Provides the mapping from the Swift type to the underlying `DataType`.
1313
@available(*, deprecated, message: "This API is experimental.")
14-
protocol Scalar {
14+
public protocol Scalar {
1515
/// The `DataType` corresponding to this scalar type.
1616
static var dataType: DataType { get }
1717
}
1818

1919
@available(*, deprecated, message: "This API is experimental.")
20-
extension UInt8: Scalar { static var dataType: DataType { .byte } }
20+
extension UInt8: Scalar { public static var dataType: DataType { .byte } }
2121
@available(*, deprecated, message: "This API is experimental.")
22-
extension Int8: Scalar { static var dataType: DataType { .char } }
22+
extension Int8: Scalar { public static var dataType: DataType { .char } }
2323
@available(*, deprecated, message: "This API is experimental.")
24-
extension Int16: Scalar { static var dataType: DataType { .short } }
24+
extension Int16: Scalar { public static var dataType: DataType { .short } }
2525
@available(*, deprecated, message: "This API is experimental.")
26-
extension Int32: Scalar { static var dataType: DataType { .int } }
26+
extension Int32: Scalar { public static var dataType: DataType { .int } }
2727
@available(*, deprecated, message: "This API is experimental.")
28-
extension Int64: Scalar { static var dataType: DataType { .long } }
28+
extension Int64: Scalar { public static var dataType: DataType { .long } }
2929
@available(*, deprecated, message: "This API is experimental.")
30-
extension Int: Scalar { static var dataType: DataType { .long } }
30+
extension Int: Scalar { public static var dataType: DataType { .long } }
3131
@available(*, deprecated, message: "This API is experimental.")
32-
extension Float: Scalar { static var dataType: DataType { .float } }
32+
extension Float: Scalar { public static var dataType: DataType { .float } }
3333
@available(*, deprecated, message: "This API is experimental.")
34-
extension Double: Scalar { static var dataType: DataType { .double } }
34+
extension Double: Scalar { public static var dataType: DataType { .double } }
3535
@available(*, deprecated, message: "This API is experimental.")
36-
extension Bool: Scalar { static var dataType: DataType { .bool } }
36+
extension Bool: Scalar { public static var dataType: DataType { .bool } }
3737
@available(*, deprecated, message: "This API is experimental.")
38-
extension UInt16: Scalar { static var dataType: DataType { .uInt16 } }
38+
extension UInt16: Scalar { public static var dataType: DataType { .uInt16 } }
3939
@available(*, deprecated, message: "This API is experimental.")
40-
extension UInt32: Scalar { static var dataType: DataType { .uInt32 } }
40+
extension UInt32: Scalar { public static var dataType: DataType { .uInt32 } }
4141
@available(*, deprecated, message: "This API is experimental.")
42-
extension UInt64: Scalar { static var dataType: DataType { .uInt64 } }
42+
extension UInt64: Scalar { public static var dataType: DataType { .uInt64 } }
4343
@available(*, deprecated, message: "This API is experimental.")
44-
extension UInt: Scalar { static var dataType: DataType { .uInt64 } }
44+
extension UInt: Scalar { public static var dataType: DataType { .uInt64 } }
4545

4646
@available(*, deprecated, message: "This API is experimental.")
47-
extension Tensor {
47+
public extension Tensor {
4848
/// Calls the closure with a typed, immutable buffer pointer over the tensor’s elements.
4949
///
5050
/// - Parameter body: A closure that receives an `UnsafeBufferPointer<T>` bound to the tensor’s data.

extension/apple/ExecuTorch/__tests__/TensorTest.swift

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ class TensorTest: XCTestCase {
153153
let tensor = data.withUnsafeMutableBytes {
154154
Tensor(bytesNoCopy: $0.baseAddress!, shape: [2, 3], dataType: .float)
155155
}
156-
let array: [Float] = try tensor.withUnsafeBytes { Array($0) }
156+
let array = try tensor.withUnsafeBytes([Float].init)
157157
XCTAssertEqual(array, data)
158158
}
159159

0 commit comments

Comments
 (0)