[CICD] Improve CUDA unit test coverage #167

Job	Run time
run_tests / checkout_and_config	6s
run_tests / unit_tests (a100) / unit-a100-pytorch_debug	47m 47s
run_tests / unit_tests (a100) / unit-a100-pytorch_distributed_unittest	45m 22s
run_tests / unit_tests (a100) / unit-a100-pytorch_onnx_unittest	45m 1s
run_tests / unit_tests (a100) / unit-a100-pytorch_unittest	58m 52s
run_tests / unit_tests (a100) / unit-a100-pytorch-unittest-coverage	3m 20s
run_tests / unit_tests_complete	4s
run_tests / integration_tests (a100) / integration-a100-pytorch_mcore_integration	3m 17s
run_tests / integration_tests_complete	2s
run_tests / all_tests_complete	4s
all_tests	3s
run_tests / checkout_and_config	6s
run_tests / unit_tests (a100) / unit-a100-pytorch_onnx_unittest	45m 1s
run_tests / unit_tests (a100) / unit-a100-pytorch_debug	47m 47s
run_tests / unit_tests (a100) / unit-a100-pytorch_distributed_unittest	45m 22s
run_tests / unit_tests (a100) / unit-a100-pytorch-unittest-coverage	3m 20s
run_tests / integration_tests (a100) / integration-a100-pytorch_mcore_integration	40m 31s
run_tests / unit_tests_complete	4s
run_tests / unit_tests (a100) / unit-a100-pytorch_unittest	58m 52s
run_tests / integration_tests_complete	4s
run_tests / all_tests_complete	3s
all_tests	2s
	7h 25m 10s

Provide feedback