Skip to content

Commit 8363a93

Browse files
niuchlcopybara-github
authored andcommitted
Bring back the legacy TensorBuffer::CreateFromHostMemory implementation.
LiteRT-PiperOrigin-RevId: 826159522
1 parent 7b10b11 commit 8363a93

File tree

2 files changed

+13
-4
lines changed

2 files changed

+13
-4
lines changed

litert/cc/litert_tensor_buffer.cc

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,18 @@ Expected<TensorBuffer> TensorBuffer::CreateManaged(
6262
return TensorBuffer(tensor_buffer, OwnHandle::kYes);
6363
}
6464

65+
Expected<TensorBuffer> TensorBuffer::CreateFromHostMemory(
66+
const RankedTensorType& tensor_type, void* host_mem_addr,
67+
size_t buffer_size) {
68+
LiteRtTensorBuffer tensor_buffer;
69+
auto litert_tensor_type = static_cast<LiteRtRankedTensorType>(tensor_type);
70+
71+
LITERT_RETURN_IF_ERROR(LiteRtCreateTensorBufferFromHostMemory(
72+
&litert_tensor_type, host_mem_addr, buffer_size,
73+
/*deallocator=*/nullptr, &tensor_buffer));
74+
return TensorBuffer(tensor_buffer, OwnHandle::kYes);
75+
}
76+
6577
Expected<TensorBuffer> TensorBuffer::CreateFromHostMemory(
6678
const Environment&, const RankedTensorType& tensor_type,
6779
void* host_mem_addr, size_t buffer_size) {

litert/cc/litert_tensor_buffer.h

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -105,10 +105,7 @@ class TensorBuffer
105105
[[deprecated]]
106106
static Expected<TensorBuffer> CreateFromHostMemory(
107107
const RankedTensorType& tensor_type, void* host_mem_addr,
108-
size_t buffer_size) {
109-
LITERT_ASSIGN_OR_RETURN(auto env, Environment::Create({}));
110-
return CreateFromHostMemory(env, tensor_type, host_mem_addr, buffer_size);
111-
}
108+
size_t buffer_size);
112109

113110
// Creates a TensorBuffer object that wraps an Android Hardware Buffer. Note
114111
// that the provided AHardwareBuffer is not owned by the TensorBuffer object

0 commit comments

Comments
 (0)