saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Build fix

Browse source
This commit is contained in:
Jeff Bloomfield 2023-04-21 17:24:58 -07:00
parent 4a510e5ba4
commit fc1af00d6c
1 changed files with 13 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
onnxruntime/core/providers/dml/DmlExecutionProvider/src/MLOperatorAuthorImpl.cpp
View file

@ -1158,8 +1158,20 @@ namespace Windows::AI::MachineLearning::Adapter
{
ORT_TRY
{
ComPtr<IMLOperatorTensor> tensorWrapper = m_constantInputGetter(inputIndex);
*tensor = nullptr;
auto constantInput = m_constantInputGetter(inputIndex);
if (!std::holds_alternative<ComPtr<IMLOperatorTensor>>(constantInput))
{
assert(std::find(
m_requiredConstantCpuInputs.begin(),
m_requiredConstantCpuInputs.end(),
inputIndex) == m_requiredConstantCpuInputs.end());
return S_OK;
}
auto tensorWrapper = std::get<ComPtr<IMLOperatorTensor>>(constantInput);
if (tensorWrapper == nullptr)
{
assert(std::find(