Skip to content

Commit 59bbe6c

Browse files
committed
try to fix test_exe_fleet_model_run
1 parent 1dc7d2e commit 59bbe6c

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

cmake/external/cinn.cmake

+5-2
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ if(NOT WITH_CINN)
1717
endif()
1818

1919
if(NOT CINN_GIT_TAG)
20-
# 2023.01.12 commit
20+
# 2023.01.28 commit
2121
set(CINN_GIT_TAG 04cbe3ca3cd17a8267a229ce7ab8d8b8b63c7f4e)
2222
endif()
2323

@@ -40,7 +40,10 @@ set(CINN_OPTIONAL_ARGS
4040
-DWITH_MKL_CBLAS=${WITH_MKL}
4141
-DWITH_MKLDNN=${WITH_MKL}
4242
-DPUBLISH_LIBS=ON
43-
-DWITH_TESTING=ON)
43+
-DWITH_TESTING=ON
44+
-DPYTHON_EXECUTABLE=${PYTHON_EXECUTABLE}
45+
-DPYTHON_INCLUDE_DIR=${PYTHON_INCLUDE_DIR}
46+
-DPYTHON_LIBRARIES=${PYTHON_LIBRARIES})
4447
set(CINN_BUILD_COMMAND ${CMAKE_COMMAND} --build . --target cinnapi -j)
4548
set(CINN_BINARY_DIR ${CINN_PREFIX_DIR}/src/external_cinn-build)
4649
set(CINN_LIB_NAME "libcinnapi.so")

paddle/fluid/distributed/fleet_executor/dist_model.cc

-1
Original file line numberDiff line numberDiff line change
@@ -442,7 +442,6 @@ bool DistModel::PrepareFleetExe() {
442442
} else {
443443
rank_info->set_ip_port(config_.trainer_endpoints[i]);
444444
}
445-
rank_info->set_ip_port(config_.trainer_endpoints[i]);
446445
id_to_rank.insert({i, i});
447446
}
448447
fleet_exe.reset(new FleetExecutor(executor_desc_));

0 commit comments

Comments
 (0)