Add tensorrt_provider_factory.h to artifact (#8869)

microsoft · Aug 27, 2021 · 6a477ac · 6a477ac
1 parent 7e53a1d
commit 6a477ac
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/...ld/github/azure-pipelines/templates/c-api-artifacts-package-and-publish-steps-windows.yml b/...ld/github/azure-pipelines/templates/c-api-artifacts-package-and-publish-steps-windows.yml
@@ -58,6 +58,7 @@ steps:
           copy $(Build.SourcesDirectory)\include\onnxruntime\core\session\onnxruntime_*.h  $(Build.BinariesDirectory)\${{parameters.artifactName}}\include
           copy $(Build.SourcesDirectory)\include\onnxruntime\core\framework\provider_options.h  $(Build.BinariesDirectory)\${{parameters.artifactName}}\include
           copy $(Build.SourcesDirectory)\include\onnxruntime\core\providers\cpu\cpu_provider_factory.h  $(Build.BinariesDirectory)\${{parameters.artifactName}}\include
+          copy $(Build.SourcesDirectory)\include\onnxruntime\core\providers\tensorrt\tensorrt_provider_factory.h  $(Build.BinariesDirectory)\${{parameters.artifactName}}\include
 
           REM copy the README, licence and TPN
           copy $(Build.SourcesDirectory)\README.md $(Build.BinariesDirectory)\${{parameters.artifactName}}\README.md          

diff --git a/tools/ci_build/github/linux/extract_and_bundle_gpu_package.sh b/tools/ci_build/github/linux/extract_and_bundle_gpu_package.sh
@@ -28,3 +28,4 @@ rm $ARTIFACT_DIR/onnxruntime-linux-x64-cuda-*.tgz
 cp onnxruntime-linux-x64-tensorrt/*/lib/libonnxruntime.so* onnxruntime-linux-x64-gpu/*/lib
 cp onnxruntime-linux-x64-tensorrt/*/lib/libonnxruntime_providers_tensorrt.so onnxruntime-linux-x64-gpu/*/lib
 cp onnxruntime-linux-x64-tensorrt/*/lib/libonnxruntime_providers_shared.so onnxruntime-linux-x64-gpu/*/lib
+cp onnxruntime-linux-x64-tensorrt/*/include/*tensorrt* onnxruntime-linux-x64-gpu/*/include
diff --git a/tools/ci_build/github/windows/bundle_dlls_gpu_nuget.bat b/tools/ci_build/github/windows/bundle_dlls_gpu_nuget.bat
@@ -19,5 +19,9 @@ FOR /R %%i IN (*.nupkg) do (
        move onnxruntime-win-x64-tensorrt\lib\onnxruntime.lib runtimes\win-x64\native\onnxruntime.lib
        move onnxruntime-win-x64-tensorrt\lib\onnxruntime.pdb runtimes\win-x64\native\onnxruntime.pdb
        7z a  %%~ni.nupkg runtimes
+
+       mkdir build\native\include
+       move onnxruntime-linux-x64-tensorrt\include\tensorrt_provider_factory.h build\native\include\tensorrt_provider_factory.h 
+       7z a  %%~ni.nupkg build 
    )
 )