Skip to content

Address DML crashes in WebNN QDQ subgraph tests #26822

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Edward Chen (edgchen1) merged 4 commits into from
Dec 19, 2025
Merged

Address DML crashes in WebNN QDQ subgraph tests #26822

Changes from 2 commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
36a8c59
Candidate fix
Dec 17, 2025
c479739
Fix for minimal build
Dec 19, 2025
4ae7c57
PR feedback (use variables and restructure logic to match ordering)
Dec 19, 2025
413c2d1
Plural -> singular
Dec 19, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 15 additions & 3 deletions ...ntime/core/optimizer/qdq_transformer/selectors_actions/qdq_selector_action_transformer.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -148,18 +148,30 @@ void UnaryOpQDQRules(SelectorActionRegistry& qdq_selector_action_registry) {
std::unique_ptr<Action> action = std::make_unique<QDQ::UnaryReplaceWithQLinear>(kMSDomain);

#if !defined(ORT_MINIMAL_BUILD)
// Operators without DML QLinear kernel support (CPU only).
// DML EP does not have QLinearLeakyRelu or QLinearSoftmax kernels.
// See https://github.com/microsoft/onnxruntime/issues/26531
std::vector<const char*> cpu_only_providers = {kCpuExecutionProvider};
std::unique_ptr<NodeSelector> cpu_selector = std::make_unique<QDQ::UnarySelector>(cpu_only_providers);
qdq_selector_action_registry.RegisterSelectorAndAction("1DQ_CPU",
Comment thread
adrastogi marked this conversation as resolved.
Outdated
Comment thread
adrastogi marked this conversation as resolved.
Outdated
{{"LeakyRelu", {}},
{"Softmax", {}}},
std::move(cpu_selector),
std::make_unique<QDQ::UnaryReplaceWithQLinear>(kMSDomain));

// Operators with DML QLinear kernel support (QLinearAveragePool, QLinearGlobalAveragePool, QLinearSigmoid)
std::vector<const char*> providers = {kCpuExecutionProvider, kDmlExecutionProvider};
std::unique_ptr<NodeSelector> selector = std::make_unique<QDQ::UnarySelector>(providers);
qdq_selector_action_registry.RegisterSelectorAndAction(action_name,
{{"AveragePool", {}},
{"LeakyRelu", {}},
{"GlobalAveragePool", {}},
{"Sigmoid", {}},
{"Softmax", {}}},
{"Sigmoid", {}}},
std::move(selector),
std::move(action));
#else
qdq_selector_action_registry.RegisterAction(action_name, std::move(action));
qdq_selector_action_registry.RegisterAction("1DQ_CPU",
Comment thread
adrastogi marked this conversation as resolved.
Outdated
std::make_unique<QDQ::UnaryReplaceWithQLinear>(kMSDomain));
#endif
}

Expand Down
Loading