This function disables printing when not in master process
(is_master)
| 168 | |
| 169 | |
| 170 | def setup_for_distributed(is_master): |
| 171 | """ |
| 172 | This function disables printing when not in master process |
| 173 | """ |
| 174 | builtin_print = builtins.print |
| 175 | |
| 176 | def print(*args, **kwargs): |
| 177 | force = kwargs.pop('force', False) |
| 178 | force = force or (get_world_size() > 8) |
| 179 | if is_master or force: |
| 180 | now = datetime.datetime.now().time() |
| 181 | builtin_print('[{}] '.format(now), end='') # print with time stamp |
| 182 | builtin_print(*args, **kwargs) |
| 183 | |
| 184 | builtins.print = print |
| 185 | |
| 186 | |
| 187 | def is_dist_avail_and_initialized(): |
no outgoing calls
no test coverage detected