未验证 提交 66ad68ed 编写于 作者: H hutuxian 提交者: GitHub

Update UT test_boxps (#19599)

Disable test_boxps in win32.
Adjust filename to avoid latent multi-thread problem.
上级 f2ad30c4
...@@ -35,6 +35,10 @@ if(NOT WITH_GPU OR WIN32) ...@@ -35,6 +35,10 @@ if(NOT WITH_GPU OR WIN32)
LIST(REMOVE_ITEM TEST_OPS test_reducescatter) LIST(REMOVE_ITEM TEST_OPS test_reducescatter)
endif() endif()
if(WIN32)
LIST(REMOVE_ITEM TEST_OPS test_boxps)
endif()
LIST(REMOVE_ITEM TEST_OPS test_launch) LIST(REMOVE_ITEM TEST_OPS test_launch)
if (NOT ${WITH_GPU}) if (NOT ${WITH_GPU})
......
...@@ -58,8 +58,9 @@ class TestBoxPSPreload(unittest.TestCase): ...@@ -58,8 +58,9 @@ class TestBoxPSPreload(unittest.TestCase):
batch1 = np.ones( batch1 = np.ones(
(batch_size, 2, 1)).astype("int64").reshape(batch_size, 2, 1) (batch_size, 2, 1)).astype("int64").reshape(batch_size, 2, 1)
filelist = [] filelist = []
place_str = "cpu" if is_cpu else "gpu"
for i in range(2): for i in range(2):
filelist.append("test_hdfs_" + str(i)) filelist.append("test_hdfs_" + place_str + "_" + str(i))
for f in filelist: for f in filelist:
with open(f, "w") as fout: with open(f, "w") as fout:
for ins in batch1: for ins in batch1:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册