Skip to content

Commit 3e2d104

Browse files
authored
Add NS_NOESCAPE attributes to Tensor getters (#11032)
1 parent 06b946e commit 3e2d104

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

extension/apple/ExecuTorch/Exported/ExecuTorchTensor.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -175,7 +175,7 @@ __attribute__((deprecated("This API is experimental.")))
175175
* - the total number of elements,
176176
* - and the data type.
177177
*/
178-
- (void)bytesWithHandler:(void (^)(const void *pointer, NSInteger count, ExecuTorchDataType dataType))handler
178+
- (void)bytesWithHandler:(NS_NOESCAPE void (^)(const void *pointer, NSInteger count, ExecuTorchDataType dataType))handler
179179
NS_SWIFT_NAME(bytes(_:));
180180

181181
/**
@@ -186,7 +186,7 @@ __attribute__((deprecated("This API is experimental.")))
186186
* - the total number of elements,
187187
* - and the data type.
188188
*/
189-
- (void)mutableBytesWithHandler:(void (^)(void *pointer, NSInteger count, ExecuTorchDataType dataType))handler
189+
- (void)mutableBytesWithHandler:(NS_NOESCAPE void (^)(void *pointer, NSInteger count, ExecuTorchDataType dataType))handler
190190
NS_SWIFT_NAME(mutableBytes(_:));
191191

192192
/**

extension/apple/ExecuTorch/Exported/ExecuTorchTensor.mm

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,12 +100,12 @@ - (NSInteger)count {
100100
return _tensor->numel();
101101
}
102102

103-
- (void)bytesWithHandler:(void (^)(const void *pointer, NSInteger count, ExecuTorchDataType type))handler {
103+
- (void)bytesWithHandler:(NS_NOESCAPE void (^)(const void *pointer, NSInteger count, ExecuTorchDataType type))handler {
104104
ET_CHECK(handler);
105105
handler(_tensor->unsafeGetTensorImpl()->data(), self.count, self.dataType);
106106
}
107107

108-
- (void)mutableBytesWithHandler:(void (^)(void *pointer, NSInteger count, ExecuTorchDataType dataType))handler {
108+
- (void)mutableBytesWithHandler:(NS_NOESCAPE void (^)(void *pointer, NSInteger count, ExecuTorchDataType dataType))handler {
109109
ET_CHECK(handler);
110110
handler(_tensor->unsafeGetTensorImpl()->mutable_data(), self.count, self.dataType);
111111
}

0 commit comments

Comments
 (0)