mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
671fa60327
* switch to work PC * back with iterable of buffers * add raw api tests * tensorization * last test * all tests pass! * small cleanup * whitespace * newline * whitespace * refactor common code into DisjointBufferHelpers * remove unused file * warning * skip gpu tests when hardware not available * Add error condition when createreference is invoked * add null check to cretereference * uncomment out check Co-authored-by: Sheil Kumar <sheilk@microsoft.com>
69 lines
1.7 KiB
C++
69 lines
1.7 KiB
C++
// Copyright 2019 Microsoft Corporation. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
#ifndef WEAK_BUFFER_H
|
|
#define WEAK_BUFFER_H
|
|
|
|
#include <wrl.h>
|
|
#include <wrl/client.h>
|
|
|
|
#include <windows.storage.streams.h>
|
|
#include <robuffer.h>
|
|
|
|
namespace Microsoft { namespace AI { namespace MachineLearning { namespace Details {
|
|
|
|
template <typename T>
|
|
struct weak_buffer
|
|
: public Microsoft::WRL::RuntimeClass<
|
|
Microsoft::WRL::RuntimeClassFlags<Microsoft::WRL::WinRtClassicComMix | Microsoft::WRL::InhibitRoOriginateError>,
|
|
ABI::Windows::Storage::Streams::IBuffer,
|
|
Windows::Storage::Streams::IBufferByteAccess> {
|
|
private:
|
|
const T* m_p_begin;
|
|
const T* m_p_end;
|
|
|
|
public:
|
|
HRESULT RuntimeClassInitialize(_In_ const T* p_begin, _In_ const T* p_end) {
|
|
m_p_begin = p_begin;
|
|
m_p_end = p_end;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE get_Capacity(
|
|
UINT32 * value)
|
|
{
|
|
if (value == nullptr) {
|
|
return E_POINTER;
|
|
}
|
|
|
|
*value = static_cast<uint32_t>(m_p_end - m_p_begin) * sizeof(T);
|
|
return S_OK;
|
|
}
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE get_Length(
|
|
UINT32 * /*value*/)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE put_Length(
|
|
UINT32 /*value*/)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHOD(Buffer)(uint8_t** value)
|
|
{
|
|
if (value == nullptr) {
|
|
return E_POINTER;
|
|
}
|
|
|
|
*value = reinterpret_cast<uint8_t*>(const_cast<T*>(m_p_begin));
|
|
return S_OK;
|
|
}
|
|
};
|
|
|
|
}}}} // namespace Microsoft::AI::MachineLearning::Details
|
|
|
|
#endif // WEAK_BUFFER_H
|