test=develop, accelerate_hs_op and add prefetch with is_sparse

6 years ago · 2e38faa3fe
parent 1a8cbb6799
commit 2e38faa3fe
1 changed files with 1 additions and 3 deletions
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@ -336,9 +336,7 @@ def embedding(input,
    """

    helper = LayerHelper('embedding', **locals())
-    remote_prefetch = False
-    if os.environ.get('PADDLE_ENABLE_REMOTE_PREFETCH'):
-        remote_prefetch = True
+    remote_prefetch = is_sparse
    if remote_prefetch:
        assert is_sparse is True and is_distributed is False
    w = helper.create_parameter(