Skip to content

Commit 048596a

Browse files
mocsharpmocsharp
committed
Detect platform and platform config
Signed-off-by: Victor Chang <vicchang@nvidia.com>
1 parent 2512c12 commit 048596a

File tree

3 files changed

+67
-13
lines changed

3 files changed

+67
-13
lines changed

‎applications/endoscopy_tool_tracking/cpp/package-app.sh

Lines changed: 27 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1414
# See the License for the specific language governing permissions and
1515
# limitations under the License.
16+
set -e
1617

1718
GIT_ROOT=$(readlink -f ./$(git rev-parse --show-cdup))
1819

@@ -26,6 +27,21 @@ print_error() {
2627
echo -e "${RED}ERROR${NOCOLOR}:" $*
2728
}
2829

30+
get_host_gpu() {
31+
if ! command -v nvidia-smi >/dev/null; then
32+
print_error Y "Could not find any GPU drivers on host. Defaulting build to target dGPU/CPU stack."
33+
echo -n "dgpu"
34+
elif nvidia-smi 2>/dev/null | grep nvgpu -q; then
35+
echo -n "igpu"
36+
else
37+
echo -n "dgpu"
38+
fi
39+
}
40+
41+
get_host_arch() {
42+
echo -n "$(uname -m)"
43+
}
44+
2945
if [ ! -d $GIT_ROOT/build/endoscopy_tool_tracking ]; then
3046
print_error "Please build the Endoscopy Tool Tracking application first with the following command:"
3147
print_error "./dev_container build_and_run endoscopy_tool_tracking"
@@ -45,10 +61,17 @@ cp -f $GIT_ROOT/build/endoscopy_tool_tracking/operators/tool_tracking_postproces
4561
echo Updating application configuration...
4662
sed -e s!gxf_extensions/lstm_tensor_rt_inference/!!g -i $APP_PATH/endoscopy_tool_tracking.yaml
4763

64+
PLATFORM=x64-workstation
65+
GPU=$(get_host_gpu)
66+
if [ $(get_host_arch) == "aarch64" ]; then
67+
PLATFORM=igx-orin-devkit
68+
fi
4869

4970
echo -e "done\n"
50-
echo -e Use the following commands to package and run the Endoscopy Tool Tracking application:\n
51-
echo -e "Package the application:\n"
52-
echo -e "${YELLOW}holoscan package -c $APP_PATH/endoscopy_tool_tracking.yaml --platform x64-workstation -t holohub-endoscopy-tool-tracking-cpp $APP_PATH/endoscopy_tool_tracking --include onnx holoviz${NOCOLOR}"
53-
echo -e "Run the application:\n"
71+
echo -e Use the following commands to package and run the Endoscopy Tool Tracking application:
72+
echo -e "Package the application:"
73+
echo -e "${YELLOW}holoscan package -c $APP_PATH/endoscopy_tool_tracking.yaml --platform [igx-orin-devkit | jetson-agx-orin-devkit | sbsa, x64-workstation] --platform-config [igpu | dgpu] -t holohub-endoscopy-tool-tracking-cpp $APP_PATH/endoscopy_tool_tracking --include onnx holoviz${NOCOLOR}"
74+
echo -e "\nFor example:"
75+
echo -e "${YELLOW}holoscan package -c $APP_PATH/endoscopy_tool_tracking.yaml --platform ${PLATFORM} --platform-config ${GPU} -t holohub-endoscopy-tool-tracking-cpp $APP_PATH/endoscopy_tool_tracking --include onnx holoviz${NOCOLOR}"
76+
echo -e "\nRun the application:"
5477
echo -e "${YELLOW}holoscan run -r \$(docker images | grep "holohub-endoscopy-tool-tracking-cpp" | awk '{print \$1\":\"\$2}') -i $GIT_ROOT/data/endoscopy${NOCOLOR}"

‎applications/endoscopy_tool_tracking/python/package-app.sh

Lines changed: 27 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1414
# See the License for the specific language governing permissions and
1515
# limitations under the License.
16+
set -e
1617

1718
GIT_ROOT=$(readlink -f ./$(git rev-parse --show-cdup))
1819

@@ -26,6 +27,21 @@ print_error() {
2627
echo -e "${RED}ERROR${NOCOLOR}:" $*
2728
}
2829

30+
get_host_gpu() {
31+
if ! command -v nvidia-smi >/dev/null; then
32+
print_error Y "Could not find any GPU drivers on host. Defaulting build to target dGPU/CPU stack."
33+
echo -n "dgpu"
34+
elif nvidia-smi 2>/dev/null | grep nvgpu -q; then
35+
echo -n "igpu"
36+
else
37+
echo -n "dgpu"
38+
fi
39+
}
40+
41+
get_host_arch() {
42+
echo -n "$(uname -m)"
43+
}
44+
2945
if [ ! -d $GIT_ROOT/build/endoscopy_tool_tracking ]; then
3046
print_error "Please build the Endoscopy Tool Tracking application first with the following command:"
3147
print_error "./dev_container build_and_run endoscopy_tool_tracking"
@@ -45,10 +61,17 @@ cp -rf $GIT_ROOT/build/endoscopy_tool_tracking/python/lib/holohub $APP_PATH
4561
echo Updating application configuration...
4662
sed -e s!gxf_extensions/lstm_tensor_rt_inference/!!g -i $APP_PATH/endoscopy_tool_tracking.yaml
4763

64+
PLATFORM=x64-workstation
65+
GPU=$(get_host_gpu)
66+
if [ $(get_host_arch) == "aarch64" ]; then
67+
PLATFORM=igx-orin-devkit
68+
fi
4869

4970
echo -e "done\n"
50-
echo -e Use the following commands to package and run the Endoscopy Tool Tracking application:\n
51-
echo -e "Package the application:\n"
52-
echo -e "${YELLOW}holoscan package -c $APP_PATH/endoscopy_tool_tracking.yaml --platform x64-workstation -t holohub-endoscopy-tool-tracking-python $APP_PATH/endoscopy_tool_tracking.py --include onnx holoviz${NOCOLOR}"
53-
echo -e "Run the application:\n"
71+
echo -e Use the following commands to package and run the Endoscopy Tool Tracking application:
72+
echo -e "Package the application:"
73+
echo -e "${YELLOW}holoscan package -c $APP_PATH/endoscopy_tool_tracking.yaml --platform [igx-orin-devkit | jetson-agx-orin-devkit | sbsa, x64-workstation] --platform-config [igpu | dgpu] -t holohub-endoscopy-tool-tracking-python $APP_PATH/endoscopy_tool_tracking.py --include onnx holoviz${NOCOLOR}"
74+
echo -e "\nFor example:"
75+
echo -e "${YELLOW}holoscan package -c $APP_PATH/endoscopy_tool_tracking.yaml --platform ${PLATFORM} --platform-config ${GPU} -t holohub-endoscopy-tool-tracking-python $APP_PATH/endoscopy_tool_tracking.py --include onnx holoviz${NOCOLOR}"
76+
echo -e "\nRun the application:"
5477
echo -e "${YELLOW}holoscan run -r \$(docker images | grep "holohub-endoscopy-tool-tracking-python" | awk '{print \$1\":\"\$2}') -i $GIT_ROOT/data/endoscopy${NOCOLOR}"

‎applications/object_detection_torch/package-app.sh

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1414
# See the License for the specific language governing permissions and
1515
# limitations under the License.
16+
set -e
1617

1718
GIT_ROOT=$(readlink -f ./$(git rev-parse --show-cdup))
1819

@@ -37,11 +38,18 @@ echo Creating application directory $APP_PATH...
3738
mkdir -p $APP_PATH
3839
echo Copying application files to $APP_PATH...
3940
cp -f $GIT_ROOT/build/object_detection_torch/applications/object_detection_torch/object_detection_torch $APP_PATH
40-
cp -f $GIT_ROOT/build/object_detection_torch/applications/object_detection_torch/object_detection_torch.yaml $APP_PATH
41+
42+
PLATFORM=x64-workstation
43+
GPU=$(get_host_gpu)
44+
if [ $(get_host_arch) == "aarch64" ]; then
45+
PLATFORM=igx-orin-devkit
46+
fi
4147

4248
echo -e "done\n"
43-
echo -e Use the following commands to package and run the Object Detection Torch application:\n
44-
echo -e "Package the application:\n"
45-
echo -e "${YELLOW}holoscan package -c $APP_PATH/object_detection_torch.yaml --platform x64-workstation -t holohub-object-detection-torch $APP_PATH/object_detection_torch --include onnx holoviz torch${NOCOLOR}"
46-
echo -e "Run the application:\n"
49+
echo -e Use the following commands to package and run the Object Detection Torch application:
50+
echo -e "Package the application:"
51+
echo -e "${YELLOW}holoscan package -c $APP_PATH/object_detection_torch.yaml --platform [igx-orin-devkit | jetson-agx-orin-devkit | sbsa, x64-workstation] --platform-config [igpu | dgpu] -t holohub-object-detection-torch $APP_PATH/object_detection_torch --include onnx holoviz torch${NOCOLOR}"
52+
echo -e "\nFor example:"
53+
echo -e "${YELLOW}holoscan package -c $APP_PATH/object_detection_torch.yaml --platform ${PLATFORM} --platform-config ${GPU} -t holohub-object-detection-torch $APP_PATH/object_detection_torch --include onnx holoviz torch${NOCOLOR}"
54+
echo -e "\nRun the application:"
4755
echo -e "${YELLOW}holoscan run -r \$(docker images | grep "holohub-object-detection-torch" | awk '{print \$1\":\"\$2}') -i $GIT_ROOT/data/object_detection_torch${NOCOLOR}"

0 commit comments

Comments
 (0)