saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-31 23:27:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Missing logic for cuda nuget package (#7911)

Browse source
This commit is contained in:
Ryan Hill 2021-06-01 22:33:04 -07:00 committed by GitHub
parent 38ca0f4839
commit 3d734a1cdc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tools/nuget/generate_nuspec_for_native_nuget.py
View file

@ -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('<file src=' + '"' + os.path.join(args.native_build_path,
nuget_dependencies['providers_shared_lib']) +
runtimes_target + args.target_architecture + '\\native" />')