updating mock. It now wraps the calls to sync and reads config from configuration file. I believe it's better not to use the preprocessor directive, i.e. not to put any test code in the engine_tcp. I just call the mock in the test_allreduce file. It's a file purely for testing purposes, so it's fine to use the mock there.
13 lines
305 B
Plaintext
13 lines
305 B
Plaintext
# Test Case example config
|
|
# You configure which methods should fail
|
|
# Format <round>_<rank> = <operation>
|
|
# <operation> can be one of the following = allreduce, broadcast, loadcheckpoint, checkpoint
|
|
|
|
1_0 = allreduce
|
|
1_1 = broadcast
|
|
1_2 = loadcheckpoint
|
|
1_3 = checkpoint
|
|
|
|
2_0 = allreduce
|
|
2_2 = checkpoint
|