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

fix hapi fleet bug in static mode (#34311)

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