From 3d734a1cdcd3c45cf8f37302e70d58ca6e56f18a Mon Sep 17 00:00:00 2001 From: Ryan Hill <38674843+RyanUnderhill@users.noreply.github.com> Date: Tue, 1 Jun 2021 22:33:04 -0700 Subject: [PATCH] Missing logic for cuda nuget package (#7911) --- tools/nuget/generate_nuspec_for_native_nuget.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/nuget/generate_nuspec_for_native_nuget.py b/tools/nuget/generate_nuspec_for_native_nuget.py index 60e600d129..b8a5afc95f 100644 --- a/tools/nuget/generate_nuspec_for_native_nuget.py +++ b/tools/nuget/generate_nuspec_for_native_nuget.py @@ -25,7 +25,7 @@ def parse_arguments(): parser.add_argument("--is_release_build", required=False, default=None, type=str, help="Flag indicating if the build is a release build. Accepted values: true/false.") parser.add_argument("--execution_provider", required=False, default='None', type=str, - choices=['dnnl', 'openvino', 'tensorrt', 'None'], + choices=['cuda', 'dnnl', 'openvino', 'tensorrt', 'None'], help="The selected execution provider for this build.") return parser.parse_args() @@ -359,7 +359,7 @@ def generate_files(list, args): nuget_dependencies['openvino_ep_shared_lib']) + runtimes_target + args.target_architecture + '\\native" />') - if args.execution_provider == "cuda": + if args.execution_provider == "cuda" or is_cuda_gpu_package: files_list.append('')