Atul Kulkarni
0c2c61d2f1
* Added python test runner to execute rccl tests * Disabled capture output to avoid hangs * Add RCCL_TEST_MPI_HOSTFILE env var to get the hostfile * Converted test_type to boolean gtest flag * Removed unused return values * Added custom rccl library usage * Removed json output * Updates to test_runner: added num_gpus field * Address review comments * Prepend env vars for single node, single process executions * Added separate enums for exit and result codes * Update configuration files * Moved configurations to its own dir * Address review comments * Update tools/scripts/test_runner/README.md Co-authored-by: Corey Derochie <161367113+corey-derochie-amd@users.noreply.github.com> --------- Co-authored-by: Corey Derochie <161367113+corey-derochie-amd@users.noreply.github.com>
21 行
448 B
Python
21 行
448 B
Python
"""
|
|
RCCL Test Runner Library
|
|
Provides modules for test configuration, parsing, and execution
|
|
"""
|
|
|
|
from .test_config import TestConfigProcessor
|
|
from .test_parser import ArgumentParserInterface, parse_test_output
|
|
from .test_executor import TestExecutor, ExitCode, TestResult
|
|
|
|
__all__ = [
|
|
'TestConfigProcessor',
|
|
'ArgumentParserInterface',
|
|
'parse_test_output',
|
|
'TestExecutor',
|
|
'ExitCode',
|
|
'TestResult'
|
|
]
|
|
|
|
__version__ = '1.0.0'
|
|
|