mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-25 22:26:24 +00:00
5350abe19d
this is a big PR. we are going to move it up to layer_dev , which is still a L3 so we are still safe to do work there agile. we are going to move this into the L3 so that ryan can start doing intergration testing. we will pause for a full code review and integration test result prior to going into the L2. >>>> raw comments from previous commits >>> * LearningModelSession is cleaned up to use the adapter, and parts of binding are. * moved everything in the winmladapter made it all nano-com using, WRL to construct objects in the ORT side. base interfaces for everythign for winml to call cleaned up a bunch of winml to use the base interfaces. * more pieces * GetData across the abi. * renamed some namepsace cleaned up OrtValue cleaned up Tensor cleaned up custom ops. everything *but* learnignmodel should be clean * make sure it's building. winml.dll is still a monolith.
77 lines
2.4 KiB
C++
77 lines
2.4 KiB
C++
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "pch.h"
|
|
|
|
#include "ZeroCopyInputStreamWrapper.h"
|
|
|
|
#include "winrt/Windows.Foundation.h"
|
|
|
|
using namespace Windows::AI::MachineLearning;
|
|
|
|
// ZeroCopyInputStreamWrapper
|
|
ZeroCopyInputStreamWrapper::ZeroCopyInputStreamWrapper(
|
|
ABI::Windows::Storage::Streams::IRandomAccessStreamReference* stream) {
|
|
winrt::copy_from_abi(stream_, (void*)stream);
|
|
}
|
|
|
|
bool ZeroCopyInputStreamWrapper::Next(
|
|
const void** data,
|
|
int* size) {
|
|
if (finished_reading_) {
|
|
return false;
|
|
}
|
|
|
|
auto content = stream_.OpenReadAsync().get();
|
|
|
|
wss::Buffer buffer(static_cast<uint32_t>(content.Size()));
|
|
auto result = content.ReadAsync(
|
|
buffer,
|
|
buffer.Capacity(),
|
|
wss::InputStreamOptions::None)
|
|
.get();
|
|
|
|
bytes_ = buffer.try_as<::Windows::Storage::Streams::IBufferByteAccess>();
|
|
#ifdef LAYERING_DONE
|
|
WINML_THROW_HR_IF_NULL_MSG(E_UNEXPECTED, bytes_, "Model stream is invalid.");
|
|
WINML_THROW_IF_FAILED_MSG(
|
|
bytes_->Buffer(reinterpret_cast<byte**>(const_cast<void**>(data))),
|
|
"Failed to acquire buffer from model stream.");
|
|
#else
|
|
bytes_->Buffer(reinterpret_cast<byte**>(const_cast<void**>(data)));
|
|
#endif
|
|
|
|
*size = static_cast<uint32_t>(content.Size());
|
|
finished_reading_ = true;
|
|
return true;
|
|
}
|
|
|
|
// BackUp is used when parsing encounters an error and needs to move
|
|
// back to the beginning of the erroneous chunk. We don't support random access,
|
|
// so we don't have a pointer to move back, but this can also happen for
|
|
// decrypted strings since they can have extra memory at the end that
|
|
// isn't valid. We don't want to parse non-model related data so we
|
|
// don't support this. I'd like to thrown an error here, but protobuf would
|
|
// eat that error and terminate the app. So instead we do nothing and handle
|
|
// this in LoadFromStream when the protobuf parsing returns false.
|
|
void ZeroCopyInputStreamWrapper::BackUp(int count) {
|
|
// purposely do nothing.
|
|
}
|
|
|
|
// the following methods are required by the interface,
|
|
// but they aren't actually used by ModelProto parse code,
|
|
bool ZeroCopyInputStreamWrapper::Skip(
|
|
int count) {
|
|
#ifdef LAYERING_DONE
|
|
WINML_THROW_HR(E_NOTIMPL);
|
|
#endif
|
|
return false;
|
|
}
|
|
|
|
__int64
|
|
ZeroCopyInputStreamWrapper::ByteCount() const {
|
|
#ifdef LAYERING_DONE
|
|
WINML_THROW_HR(E_NOTIMPL);
|
|
#endif
|
|
return 0;
|
|
}
|