Skip to content

Commit 5f1a6bb

Browse files
committed
test tf with correct logging
1 parent 87c9a88 commit 5f1a6bb

File tree

3 files changed

+9
-9
lines changed

3 files changed

+9
-9
lines changed

dlc_developer_config.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ deep_canary_mode = false
3737
[build]
3838
# Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image.
3939
# available frameworks - ["base", "vllm", "autogluon", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"]
40-
build_frameworks = ["pytorch"]
40+
build_frameworks = ["tensorflow"]
4141

4242

4343
# By default we build both training and inference containers. Set true/false values to determine which to build.

test/dlc_tests/container_tests/bin/efa/testEFA

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,10 +100,10 @@ check_efa_nccl_all_reduce(){
100100
check_efa_nccl_all_reduce_performance
101101
}
102102

103-
check_efa_nccl_all_reduce
104-
105103
nvidia-smi | tee /test/logs/smi.log
106104
RESULT=$(cat /test/logs/smi.log)
107105

108106
nvcc --version | tee /test/logs/nvcc.log
109107
RESULT=$(cat /test/logs/nvcc.log)
108+
109+
check_efa_nccl_all_reduce

test/dlc_tests/container_tests/bin/efa/testEFASanity

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,22 +6,22 @@ set -ex
66
export PATH=/opt/amazon/efa/bin:$PATH
77

88
# check if efa components are correctly installed
9-
cat /opt/amazon/efa_installed_packages
9+
RESULT=$(cat /opt/amazon/efa_installed_packages)
1010

1111
# check Libfabric EFA interfaces
12-
fi_info -p efa
13-
fi_info -p efa -t FI_EP_RDM | grep 'FI_EP_RDM'
12+
RESULT=$(fi_info -p efa)
13+
RESULT=$(fi_info -p efa -t FI_EP_RDM | grep 'FI_EP_RDM')
1414

1515
apt-get update && apt-get install -y kmod
1616

1717
# check if ib_uverbs is present
18-
lsmod | grep ib_uverbs
18+
RESULT=$(lsmod | grep ib_uverbs)
1919

2020
# ensure that the security group created is configured correctly
2121
/test/bin/efa/efa_test.sh
2222

2323
# Queries local RDMA devices
24-
ibv_devinfo
24+
RESULT=$(ibv_devinfo)
2525

2626
# check if gdr device is loaded
27-
cat /sys/class/infiniband/**/device/p2p | grep 'NVIDIA'
27+
RESULT=$(cat /sys/class/infiniband/**/device/p2p | grep 'NVIDIA')

0 commit comments

Comments
 (0)