commit 2b79e393560d545c289e8164805c52bf874838a4
author Ben Vanik <benvanik@google.com> Fri Jun 18 12:15:25 2021 -0700
committer GitHub <noreply@github.com> Fri Jun 18 12:15:25 2021 -0700
tree 037034909e1490a5c38e2b497ad20f089368f666
parent 0c80a86995547eb83baddc94eb7cae96ba398484

Removing the use of hal.ex.shared_device from tensorlist. (#6241)

iree/compiler/Dialect/HAL/Utils/TypeUtils.cpp

diff --git a/iree/compiler/Dialect/HAL/Utils/TypeUtils.cpp b/iree/compiler/Dialect/HAL/Utils/TypeUtils.cpp
index bc2937a..0c923bf 100644
--- a/iree/compiler/Dialect/HAL/Utils/TypeUtils.cpp
+++ b/iree/compiler/Dialect/HAL/Utils/TypeUtils.cpp
@@ -103,9 +103,8 @@
   if (!elementType) return {};
   auto shape = IREE::HAL::getShapeDims(loc, value, builder);
   if (!shape) return {};
-  auto deviceValue = builder.createOrFold<IREE::HAL::ExSharedDeviceOp>(loc);
-  auto allocatorValue =
-      builder.createOrFold<IREE::HAL::DeviceAllocatorOp>(loc, deviceValue);
+  auto allocatorValue = builder.createOrFold<IREE::HAL::BufferAllocatorOp>(
+      loc, IREE::HAL::AllocatorType::get(builder.getContext()), value);
   return builder.createOrFold<IREE::HAL::AllocatorComputeSizeOp>(
       loc, allocatorValue, *shape, elementType.getValue());
 }