def _register_stream(stream):
if not _initialized:
raise RuntimeError("torch.distributed needs to be initialized first")
return torch._C._dist_register_stream(stream)
微信
公众号
扫码关注公众号