From 2e38faa3fe279520f98b2030e35ae8db68ba66d8 Mon Sep 17 00:00:00 2001 From: JiabinYang Date: Mon, 24 Dec 2018 10:17:32 +0000 Subject: [PATCH] test=develop, accelerate_hs_op and add prefetch with is_sparse --- python/paddle/fluid/layers/nn.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index 6379031ee..9af62bf06 100644 --- 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( -- GitLab