Skip to content

Commit 05909c6

Browse files
authored
Update ExecuTorchTensor.mm
1 parent 4dc82b9 commit 05909c6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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)