Fix unsafe buffer usage warning (#114)

PTAL /cc @shiyi9801
shiyi9801 · Feb 11, 2025 · 44a7c7f · 44a7c7f
1 parent 961d1c0
commit 44a7c7f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/services/webnn/ort/tensor_impl_ort.cc b/services/webnn/ort/tensor_impl_ort.cc
@@ -99,9 +99,11 @@ void TensorImplOrt::ReadTensorImpl(ReadTensorCallback callback) {
                 buffer_state->GetSharedLockedResource().tensor(),
                 &ort_tensor_raw_data));
             CHECK(ort_tensor_raw_data);
-            mojo_base::BigBuffer output_buffer(
+            // SAFETY: ORT guarantees that it has allocated enough memory to
+            // store tensor.
+            mojo_base::BigBuffer output_buffer(UNSAFE_BUFFERS(
                 base::span(static_cast<const uint8_t*>(ort_tensor_raw_data),
-                           bytes_to_read));
+                           bytes_to_read)));
 
             // Unlock the buffer contents.
             std::move(completion_closure).Run();