1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
|
#!/bin/bash
echo "# Running Python Testing Programs"
echo "# The tests are collected from test/run_test.py"
FILES=(
benchmark_utils/test_benchmark_utils
distributed/test_c10d
distributed/test_c10d_spawn
distributed/test_data_parallel
distributed/test_distributed_fork
distributed/test_distributed_spawn
distributed/test_jit_c10d
distributed/test_nccl
distributions/test_constraints
distributions/test_distributions
test_autograd
test_binary_ufuncs
test_bundled_inputs
test_complex
test_cpp_api_parity
test_cpp_extensions_aot_ninja
test_cpp_extensions_aot_no_ninja
test_cpp_extensions_jit
test_cuda
test_cuda_primary_ctx
test_dataloader
test_datapipe
test_dataset
test_dispatch
test_expecttest
test_foreach
test_indexing
test_jit
test_jit_cuda_fuser
test_linalg
test_logging
test_mkldnn
test_mobile_optimizer
test_multiprocessing
test_multiprocessing_spawn
test_native_functions
test_nn
test_numba_integration
test_ops
test_optim
test_pruning_op
test_pytree
test_quantization
test_serialization
test_sparse
test_spectral_ops
test_type_hints
test_vulkan
test_xnnpack_integration
)
echo "# Found" ${#FILES[@]} "tests"
echo "#"
sleep 1
failed=( )
for (( i = 0; i < ${#FILES[@]}; i++ )); do
echo
echo
echo "# Py test ${i}/${#FILES[@]} ${FILES[$i]}"
python3 -m pytest ${FILES[$i]}.py -v
if test 0 != $?; then
failed+=( ${Files[$i]} )
fi
done
echo
echo "# listing failed tests ..."
for i in ${failed[@]}; do
echo ${i}
done
|