提交 2e38faa3 编写于 作者: J JiabinYang

test=develop, accelerate_hs_op and add prefetch with is_sparse

上级 1a8cbb67
...@@ -336,9 +336,7 @@ def embedding(input, ...@@ -336,9 +336,7 @@ def embedding(input,
""" """
helper = LayerHelper('embedding', **locals()) helper = LayerHelper('embedding', **locals())
remote_prefetch = False remote_prefetch = is_sparse
if os.environ.get('PADDLE_ENABLE_REMOTE_PREFETCH'):
remote_prefetch = True
if remote_prefetch: if remote_prefetch:
assert is_sparse is True and is_distributed is False assert is_sparse is True and is_distributed is False
w = helper.create_parameter( w = helper.create_parameter(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册