未验证 提交 e05540f7 编写于 作者: S Steffy-zxf 提交者: GitHub

fix data parallel when VOCAB var in program (#37543)

* fix data parallel when VOCAB var in program
上级 6c970f9f
......@@ -370,6 +370,8 @@ def sync_params_buffers(model,
# such as moe's expert parameters
if getattr(param, "no_sync", False):
continue
if param.type == core.VarDesc.VarType.VOCAB:
continue
model_vars.append(param.detach())
if len(model_vars) == 0:
......
......@@ -578,6 +578,7 @@ py_test_modules(test_imperative_static_runner_mnist MODULES test_imperative_stat
py_test_modules(test_imperative_static_runner_while MODULES test_imperative_static_runner_while ENVS
FLAGS_cudnn_deterministic=1)
set_tests_properties(test_conv2d_op PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
set_tests_properties(test_faster_tokenizer_op PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
set_tests_properties(test_conv2d_op_depthwise_conv PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
set_tests_properties(test_conv2d_api PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
set_tests_properties(test_conv_nn_grad PROPERTIES LABELS "RUN_TYPE=EXCLUSIVE")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册