Skip to content

Commit

Permalink
Print more testing info (#190)
Browse files Browse the repository at this point in the history
  • Loading branch information
Alberto Alvarez authored Apr 6, 2022
1 parent 822c6bb commit f7d24f3
Showing 1 changed file with 20 additions and 20 deletions.
40 changes: 20 additions & 20 deletions ci/test_container.sh
Original file line number Diff line number Diff line change
Expand Up @@ -5,70 +5,70 @@ devices=$2



#print("##################")
#print("# Software check #")
#print("##################")
echo "##################"
echo "# Software check #"
echo "##################"

regex="merlin(.)*-inference"
if [[ ! "$container" =~ $regex ]]; then
# print("Check tritonserver for inference containers")
echo "Check tritonserver for inference containers"
whereis tritonserver
fi

if [ "$container" == "merlin-training" ]; then
# print("Check HugeCTR for ctr-training container")
echo "Check HugeCTR for ctr-training container"
python -c "import hugectr; print(hugectr.__version.__)"
fi

if [ "$container" == "merlin-tensorflow-training" ]; then
# print("Check TensorFlow for tf-training container")
echo "Check TensorFlow for tf-training container"
python -c "import tensorflow; print(tensorflow.__version__)"
fi

if [ "$container" == "merlin-pytorch-training" ]; then
# print("Check PyTorch for torch-training container")
echo "Check PyTorch for torch-training container"
python -c "import torch; print(torch.__version__)"
fi

#print("##############")
#print("# Unit tests #")
#print("##############")
echo "##############"
echo "# Unit tests #"
echo "##############"

## Test Core
#print("Run unit tests for Core")
echo "Run unit tests for Core"
/core/ci/test_unit.sh $container $devices

## Test NVTabular
#print("Run unit tests for NVTabular")
echo "Run unit tests for NVTabular"
/nvtabular/ci/test_unit.sh $container $devices

## Test Transformers4Rec
#print("Run unit tests for Transformers4Rec")
echo "Run unit tests for Transformers4Rec"
/transformers4rec/ci/test_unit.sh $container $devices

## Test Models
#print("Run unit tests for Models")
echo "Run unit tests for Models"
pip install coverage
/models/ci/test_unit.sh $container $devices

## Test HugeCTR
if [ "$container" == "merlin-training" ]; then
# print("Run unit tests for HugeCTR")
echo "Run unit tests for HugeCTR"
/hugectr/ci/test_unit.sh $container $devices
fi

#print("#####################")
#print("# Integration tests #")
#print("#####################")
echo "#####################"
echo "# Integration tests #"
echo "#####################"

# Test NVTabular
## Not shared storage in blossom yet, inference testing cannot be run
regex="merlin(.)*-inference"
if [[ ! "$container" =~ $regex ]]; then
# print("Run instegration tests for NVTabular")
echo "Run instegration tests for NVTabular"
/nvtabular/ci/test_integration.sh $container $devices --report 1
fi

# Test Transformers4Rec
#print("Run integration tests for Transformers4Rec")
echo "Run integration tests for Transformers4Rec"
/transformers4rec/ci/test_integration.sh $container $devices

0 comments on commit f7d24f3

Please sign in to comment.