From f57515aae40044fd9c9f2b7acb4c7a7a307313a8 Mon Sep 17 00:00:00 2001
From: Jeff Bloomfield <jeffbloo@ntdev.microsoft.com>
Date: Thu, 14 May 2020 12:00:15 -0700
Subject: [PATCH] Update DML nuget to 2.1.0

---
 cmake/external/dml.cmake                        | 2 +-
 packages.config                                 | 2 +-
 tools/nuget/generate_nuspec_for_native_nuget.py | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/cmake/external/dml.cmake b/cmake/external/dml.cmake
index 0ae5277515..118aba78a6 100644
--- a/cmake/external/dml.cmake
+++ b/cmake/external/dml.cmake
@@ -20,7 +20,7 @@ if (NOT onnxruntime_USE_CUSTOM_DIRECTML)
   set(NUGET_CONFIG ${PROJECT_SOURCE_DIR}/../NuGet.config)
   set(PACKAGES_CONFIG ${PROJECT_SOURCE_DIR}/../packages.config)
   get_filename_component(PACKAGES_DIR ${CMAKE_CURRENT_BINARY_DIR}/../packages ABSOLUTE)
-  set(DML_PACKAGE_DIR ${PACKAGES_DIR}/DirectML.0.0.4)
+  set(DML_PACKAGE_DIR ${PACKAGES_DIR}/DirectML.2.1.0)
 
   # Restore nuget packages, which will pull down the DirectML redist package
   add_custom_command(
diff --git a/packages.config b/packages.config
index 479e7d9fd0..816872da32 100644
--- a/packages.config
+++ b/packages.config
@@ -1,5 +1,5 @@
 <?xml version="1.0" encoding="utf-8"?>
 <packages>
-  <package id="DirectML" version="0.0.4" targetFramework="native" />
+  <package id="DirectML" version="2.1.0" targetFramework="native" />
   <package id="GoogleTestAdapter" version="0.17.1" targetFramework="net46" />
 </packages>
diff --git a/tools/nuget/generate_nuspec_for_native_nuget.py b/tools/nuget/generate_nuspec_for_native_nuget.py
index b3f45b9a8d..896dd3bfd9 100644
--- a/tools/nuget/generate_nuspec_for_native_nuget.py
+++ b/tools/nuget/generate_nuspec_for_native_nuget.py
@@ -150,7 +150,7 @@ def generate_files(list, args):
     if includes_directml:
         files_list.append('<file src=' + '"' + os.path.join(args.native_build_path, 'DirectML.dll') + '" target="runtimes\\win-' + args.target_architecture + '\\native" />')
         files_list.append('<file src=' + '"' + os.path.join(args.native_build_path, 'DirectML.pdb') + '" target="runtimes\\win-' + args.target_architecture + '\\native" />')
-        files_list.append('<file src=' + '"' + os.path.join(args.packages_path, 'DirectML.0.0.2\\LICENSE.txt') + '" target="DirectML_LICENSE.txt" />')
+        files_list.append('<file src=' + '"' + os.path.join(args.packages_path, 'DirectML.2.1.0\\LICENSE.txt') + '" target="DirectML_LICENSE.txt" />')
 
     if includes_winml:
         # Process microsoft.ai.machinelearning import lib, dll, and pdb