未验证 提交 13991b5e 编写于 作者: J Jiaqi Liu 提交者: GitHub

fix hapi fleet bug in static mode (#34311)

上级 3ea3e90d
......@@ -218,8 +218,6 @@ def prepare_distributed_context(place=None):
fluid.disable_dygraph()
_init_context()
fluid.enable_dygraph(place)
else:
_init_context()
else:
assert ("Only support CUDAPlace for now.")
......
......@@ -59,6 +59,7 @@ def compute_accuracy(pred, gt):
'CPU testing is not supported')
class TestDistTraning(unittest.TestCase):
def test_static_multiple_gpus(self):
paddle.enable_static()
device = set_device('gpu')
im_shape = (-1, 1, 28, 28)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册