upate test script for new ctr-base and merlin-hugectr

ci/container_hugectr.sh

@@ -0,0 +1,20 @@
+#!/bin/bash
+
+container=$1
+devices=$2
+
+echo "##############"
+echo "# Unit tests #"
+echo "##############"
+
+exit_code=0
+
+## Test HugeCTR
+if [ "$container" == "merlin-hugectr" ]; then
+    echo "Run unit tests for HugeCTR"
+    /hugectr/ci/test_unit.sh $container $devices || exit_code=1
+    echo "Run unit tests for merlin-sok"
+    /hugectr/ci/test_unit.sh "merlin-tensorflow" $devices || exit_code=1
+fi
+
+exit $exit_code

ci/test_container.sh

@@ -16,7 +16,12 @@ if [ $container != 'merlin-ci-runner' ]; then
     ${ci_script_dir}container_size.sh $container $devices
 fi
 
-${ci_script_dir}container_software.sh $container $devices
-${ci_script_dir}container_integration.sh $container $devices $suppress_failures
-${ci_script_dir}container_unit.sh $container $devices
+${ci_script_dir}ciontainer_software.sh $container $devices
+
+if [ $container == 'merlin-hugectr' ]; then
+    ${ci_script_dir}container_hugectr.sh $container $devices
+elif [ $container != 'ctr-base' ]; then
+    ${ci_script_dir}container_integration.sh $container $devices $suppress_failures
+    ${ci_script_dir}container_unit.sh $container $devices
+fi