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

Extend workaround with input name matching in DML fused graph kernel (#3918)

Browse source
This commit is contained in:
Jeff Bloomfield 2020-05-12 14:04:09 -07:00 committed by GitHub
parent 0f82b42fed
commit f170f31e1d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
onnxruntime/core/providers/dml/DmlExecutionProvider/src/GraphDescBuilder.cpp
View file

@ -205,6 +205,14 @@ namespace Dml::GraphDescBuilder
if (arg->Exists())
{
auto iter = nameToFusedNodeInputIndex.find(arg->Name());
// The graph input could be missing the suffix, so try to match without it.
// This is part of a temporary workaround; see comments in GetFusedNodeArgNameMatchingGraph.
if (iter == nameToFusedNodeInputIndex.end())
{
iter = nameToFusedNodeInputIndex.find(GetFusedNodeArgNameMatchingGraph(arg->Name()));
}
if (iter != nameToFusedNodeInputIndex.end())
{
// This is a graph input