From 05acfb90ab4a0bb6c434c5d18346a39b19725a8e Mon Sep 17 00:00:00 2001 From: Patrice Vignola Date: Tue, 17 Sep 2024 22:37:45 -0700 Subject: [PATCH] [DML EP] Add QDQ+MatMul fusion into MatMulNBits (#22114) ### Description ### Motivation and Context --- .../selectors_actions/qdq_selector_action_transformer.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/core/optimizer/qdq_transformer/selectors_actions/qdq_selector_action_transformer.cc b/onnxruntime/core/optimizer/qdq_transformer/selectors_actions/qdq_selector_action_transformer.cc index 1c506bafd1..71d4ddd389 100644 --- a/onnxruntime/core/optimizer/qdq_transformer/selectors_actions/qdq_selector_action_transformer.cc +++ b/onnxruntime/core/optimizer/qdq_transformer/selectors_actions/qdq_selector_action_transformer.cc @@ -290,7 +290,7 @@ void DQMatMulToMatMulNBitsRules(SelectorActionRegistry& qdq_selector_action_regi p_buffered_tensors); #if !defined(ORT_MINIMAL_BUILD) - std::vector providers = {kCpuExecutionProvider, kCudaExecutionProvider}; + std::vector providers = {kCpuExecutionProvider, kCudaExecutionProvider, kDmlExecutionProvider}; std::unique_ptr selector = std::make_unique(providers); qdq_selector_action_registry.RegisterSelectorAndAction(action_name, {{"MatMul", {}}},