github/kata-containers
1
0
Fork 1
mirror of https://github.com/kata-containers/kata-containers.git synced 2025-06-27 07:48:55 +00:00
Code Issues Projects Releases Wiki Activity

metrics: Use function from metrics common in pytorch script

Browse Source 
This PR uses a common function into the pytorch script.

Fixes #7709

Signed-off-by: Gabriela Cervantes <gabriela.cervantes.tellez@intel.com>
This commit is contained in:
Gabriela Cervantes 2023-08-21 16:03:43 +00:00
parent 700759232f
commit 39bc3488f5
1 changed files with 1 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tests/metrics/machine_learning/pytorch.sh
View File

@ -29,17 +29,6 @@ function remove_tmp_file() {
trap remove_tmp_file EXIT trap remove_tmp_file EXIT
function check_containers_are_up() {
local containers_launched=0
for i in $(seq "${TIMEOUT}") ; do
info "Verify that the containers are running"
containers_launched="$(sudo ${CTR_EXE} t list | grep -c "RUNNING")"
[ "${containers_launched}" -eq "${NUM_CONTAINERS}" ] && break
sleep 1
[ "${i}" == "${TIMEOUT}" ] && return 1
done
}
function equation_of_state_pytorch_test() { function equation_of_state_pytorch_test() {
info "Running Equation of State Pytorch test" info "Running Equation of State Pytorch test"
for i in "${containers[@]}"; do for i in "${containers[@]}"; do
@ -146,7 +135,7 @@ function main() {
# Check that the requested number of containers are running # Check that the requested number of containers are running
check_containers_are_up check_containers_are_up "${NUM_CONTAINERS}"
equation_of_state_pytorch_test equation_of_state_pytorch_test