mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-30 20:18:08 +00:00
parent
a5bef6886b
commit
2c5e603bad
1 changed files with 5 additions and 2 deletions
|
|
@ -1511,7 +1511,7 @@ def derive_linux_build_property():
|
|||
return "/p:IsLinuxBuild=\"true\""
|
||||
|
||||
|
||||
def build_nuget_package(source_dir, build_dir, configs, use_cuda, use_openvino, use_tensorrt, use_dnnl):
|
||||
def build_nuget_package(source_dir, build_dir, configs, use_cuda, use_openvino, use_tensorrt, use_dnnl, use_nuphar):
|
||||
if not (is_windows() or is_linux()):
|
||||
raise BuildError(
|
||||
'Currently csharp builds and nuget package creation is only supportted '
|
||||
|
|
@ -1534,6 +1534,8 @@ def build_nuget_package(source_dir, build_dir, configs, use_cuda, use_openvino,
|
|||
package_name = "/p:OrtPackageId=\"Microsoft.ML.OnnxRuntime.DNNL\""
|
||||
elif use_cuda:
|
||||
package_name = "/p:OrtPackageId=\"Microsoft.ML.OnnxRuntime.Gpu\""
|
||||
elif use_nuphar:
|
||||
package_name = "/p:OrtPackageId=\"Microsoft.ML.OnnxRuntime.Nuphar\""
|
||||
else:
|
||||
pass
|
||||
|
||||
|
|
@ -1960,7 +1962,8 @@ def main():
|
|||
args.use_cuda,
|
||||
args.use_openvino,
|
||||
args.use_tensorrt,
|
||||
args.use_dnnl
|
||||
args.use_dnnl,
|
||||
args.use_nuphar
|
||||
)
|
||||
|
||||
if args.test and args.build_nuget:
|
||||
|
|
|
|||
Loading…
Reference in a new issue