diff --git a/python/paddle/v2/fluid/tests/CMakeLists.txt b/python/paddle/v2/fluid/tests/CMakeLists.txt index e795627bfe9e8ad0c196349a332e62e975f20aa3..9a0240cbf65c7a79e29babc2abcb157ada684c5e 100644 --- a/python/paddle/v2/fluid/tests/CMakeLists.txt +++ b/python/paddle/v2/fluid/tests/CMakeLists.txt @@ -5,3 +5,4 @@ foreach(src ${TEST_OPS}) endforeach() add_subdirectory(book) +add_subdirectory(book_distribute) diff --git a/python/paddle/v2/fluid/tests/book_distribute/test_split_var.py b/python/paddle/v2/fluid/tests/book_distribute/test_split_var.py index 1355e13e1c11ff9cdd130ed579408d8c5416c9d9..cfb48a59154527160f622c12ae429bac31483631 100644 --- a/python/paddle/v2/fluid/tests/book_distribute/test_split_var.py +++ b/python/paddle/v2/fluid/tests/book_distribute/test_split_var.py @@ -2,6 +2,7 @@ import math import unittest from paddle.v2.fluid.distribute_transpiler import split_dense_variable import paddle.v2.fluid as fluid +import paddle.v2.fluid.core as core import random @@ -19,9 +20,9 @@ class TestSplitVar(unittest.TestCase): program = fluid.Program() for shape in shapes: var = program.global_block().create_var( - name=str(random.randint(10000)), + name=str(random.randint(10000, 99999)), persistable=True, - dtype=core.VarDesc.VarType.LOD_TENSOR, + # dtype=core.VarDesc.VarType.LOD_TENSOR, shape=shape) var_list.append(var) blocks = split_dense_variable(var_list, 10)