github/kata-containers
1
0
Fork 1
mirror of https://github.com/kata-containers/kata-containers.git synced 2025-04-29 20:24:31 +00:00
Code Issues Projects Releases Wiki Activity

metrics: Rename tensorflow scripts

Browse Source 
This PR renames the tensorflow scripts to include the data format
that is being used as we will have multiple tests with different
data and model formats for tensorflow so this will help us to
distinguish them.

Fixes #7645

Signed-off-by: Gabriela Cervantes <gabriela.cervantes.tellez@intel.com>
This commit is contained in:
Gabriela Cervantes 2023-08-14 20:36:19 +00:00
parent a740c80251
commit 18a7fd8e4e
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/metrics/machine_learning/tensorflow.sh → tests/metrics/machine_learning/tensorflow_nhwc.sh
View File

@ -10,15 +10,15 @@ set -o pipefail
SCRIPT_PATH=$(dirname "$(readlink -f "$0")") SCRIPT_PATH=$(dirname "$(readlink -f "$0")")
source "${SCRIPT_PATH}/../lib/common.bash" source "${SCRIPT_PATH}/../lib/common.bash"
IMAGE="docker.io/library/tensorflow:latest" IMAGE="docker.io/library/tensorflow_nhwc:latest"
DOCKERFILE="${SCRIPT_PATH}/tensorflow_dockerfile/Dockerfile" DOCKERFILE="${SCRIPT_PATH}/tensorflow_nhwc_dockerfile/Dockerfile"
BATCH_SIZE="100" BATCH_SIZE="100"
NUM_BATCHES="100" NUM_BATCHES="100"
resnet_tensorflow_file=$(mktemp resnettensorflowresults.XXXXXXXXXX) resnet_tensorflow_file=$(mktemp resnettensorflowresults.XXXXXXXXXX)
alexnet_tensorflow_file=$(mktemp alexnettensorflowresults.XXXXXXXXXX) alexnet_tensorflow_file=$(mktemp alexnettensorflowresults.XXXXXXXXXX)
NUM_CONTAINERS="$1" NUM_CONTAINERS="$1"
TIMEOUT="$2" TIMEOUT="$2"
TEST_NAME="tensorflow" TEST_NAME="tensorflow_nhwc"
PAYLOAD_ARGS="tail -f /dev/null" PAYLOAD_ARGS="tail -f /dev/null"
# Options to control the start of the workload using a trigger-file # Options to control the start of the workload using a trigger-file
dst_dir="/host" dst_dir="/host"
@ -99,7 +99,7 @@ function launch_workload() {
done done
} }
function tensorflow_test() { function tensorflow_nhwc_test() {
# Resnet section # Resnet section
info "Running TF-Resnet test" info "Running TF-Resnet test"
launch_workload "${CMD_RESNET}" launch_workload "${CMD_RESNET}"
@ -191,7 +191,7 @@ function main() {
# Get the initial number of pids in a single container before the workload starts # Get the initial number of pids in a single container before the workload starts
INITIAL_NUM_PIDS=$(sudo -E "${CTR_EXE}" t metrics "${containers[-1]}" | grep pids.current | grep pids.current | xargs | cut -d ' ' -f 2) INITIAL_NUM_PIDS=$(sudo -E "${CTR_EXE}" t metrics "${containers[-1]}" | grep pids.current | grep pids.current | xargs | cut -d ' ' -f 2)
((INITIAL_NUM_PIDS++)) ((INITIAL_NUM_PIDS++))
tensorflow_test tensorflow_nhwc_test
metrics_json_save metrics_json_save
} }

0
tests/metrics/machine_learning/tensorflow_dockerfile/Dockerfile → tests/metrics/machine_learning/tensorflow_nhwc_dockerfile/Dockerfile
View File