fix(session): use directml conditonally

k2-fsa · Jul 20, 2024 · 17621db · 17621db
1 parent 3e0efaa
commit 17621db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sherpa-onnx/csrc/session.cc b/sherpa-onnx/csrc/session.cc
@@ -19,7 +19,7 @@
 #include "nnapi_provider_factory.h"  // NOLINT
 #endif
 
-#if defined(_WIN32) && defined(SHERPA_ONNX_ENABLE_DIRECTML)
+#if defined(_WIN32) && SHERPA_ONNX_ENABLE_DIRECTML == 1
 #include "dml_provider_factory.h"  // NOLINT
 #endif
 
@@ -172,7 +172,7 @@ static Ort::SessionOptions GetSessionOptionsImpl(
       break;
     }
     case Provider::kDirectML: {
-#if defined(_WIN32)
+#if defined(_WIN32) && SHERPA_ONNX_ENABLE_DIRECTML == 1
       sess_opts.DisableMemPattern();
       sess_opts.SetExecutionMode(ORT_SEQUENTIAL);
       int32_t device_id = 0;