Fix iree_hal_buffer_subspan after change in iree.

nod-ai · Dec 10, 2024 · e56b568 · e56b568
1 parent 4c015d4
commit e56b568
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/shortfin/src/shortfin/array/storage.cc b/shortfin/src/shortfin/array/storage.cc
@@ -92,7 +92,9 @@ storage storage::subspan(iree_device_size_t byte_offset,
                          iree_device_size_t byte_length) {
   storage new_storage(device_, {}, timeline_resource_);
   SHORTFIN_THROW_IF_ERROR(iree_hal_buffer_subspan(
-      buffer_, byte_offset, byte_length, new_storage.buffer_.for_output()));
+      buffer_, byte_offset, byte_length,
+      iree_hal_device_host_allocator(device_.raw_device()->hal_device()),
+      new_storage.buffer_.for_output()));
   return new_storage;
 }