From c331d8cffc77aa6aebcce8581138813eda9611cb Mon Sep 17 00:00:00 2001
From: Sheil Kumar <smk2007@gmail.com>
Date: Fri, 29 May 2020 13:24:22 -0700
Subject: [PATCH] WinML custom operator header is missing from nuget package.
 (#4083)

* publish mloperatorauthor.h in the nuget

* build dmlep into arm/arm64 builds

* update to not use --use_dml everywhere, but enable custom ops everywhere

* always download directml nuget in winml builds

* always build with dml

* dont build dml for arm

Co-authored-by: Sheil Kumar <sheilk@microsoft.com>
---
 tools/nuget/generate_nuspec_for_native_nuget.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tools/nuget/generate_nuspec_for_native_nuget.py b/tools/nuget/generate_nuspec_for_native_nuget.py
index ed500a7561..ae323fa482 100644
--- a/tools/nuget/generate_nuspec_for_native_nuget.py
+++ b/tools/nuget/generate_nuspec_for_native_nuget.py
@@ -181,6 +181,11 @@ def generate_files(list, args):
         files_list.append('<file src=' + '"' + os.path.join(args.ort_build_path, args.build_config,
                                                             'microsoft.ai.machinelearning.winmd') +
                           '" target="lib\\uap10.0\\Microsoft.AI.MachineLearning.winmd" />')
+        # Add custom operator headers
+        files_list.append('<file src=' + '"' +
+                          os.path.join(args.sources_path,
+                                       'onnxruntime\\core\\providers\\dml\\dmlexecutionprovider\\inc\\mloperatorauthor.h') +
+                          '" target="build\\native\\include" />')
 
     # Process runtimes
     # Process onnxruntime import lib, dll, and pdb