saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-28 03:20:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Extend temporary enum workaround to non-compiled path

Browse source
This commit is contained in:
Jeff Bloomfield 2023-10-05 17:50:44 -07:00
parent 2efa2ab081
commit defde2cd9e
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperator.cpp
View file

@ -86,9 +86,19 @@ namespace Dml
}
else
{
auto operatorDescCopy = operatorDesc;
// TODO: Change as new header is ingested
if (operatorDescCopy.Type == (DML_OPERATOR_TYPE) DML_OPERATOR_QUANTIZED_LINEAR_AVERAGE_POOLING)
operatorDescCopy.Type = (DML_OPERATOR_TYPE) 169;
// TODO: Change as new header is ingested
if (operatorDescCopy.Type == (DML_OPERATOR_TYPE) DML_OPERATOR_MATRIX_MULTIPLY_INTEGER_TO_FLOAT)
operatorDescCopy.Type = (DML_OPERATOR_TYPE) 170;
// Create and compile the operator.
ComPtr<IDMLOperator> dmlOperator;
ORT_THROW_IF_FAILED(m_dmlDevice->CreateOperator(&operatorDesc, IID_PPV_ARGS(&dmlOperator)));
ORT_THROW_IF_FAILED(m_dmlDevice->CreateOperator(&operatorDescCopy, IID_PPV_ARGS(&dmlOperator)));
DML_EXECUTION_FLAGS executionFlags = GetExecutionFlags();
ORT_THROW_IF_FAILED(m_dmlDevice->CompileOperator(dmlOperator.Get(), executionFlags, IID_PPV_ARGS(&m_compiledOperator)));