wehub-resource-sync
97 lines
2.9 KiB
CMake
97 lines
2.9 KiB
CMake
if(WIN32)
|
|
cc_test(
|
|
device_mesh_test
|
|
SRCS device_mesh_test.cc
|
|
DEPS type_info)
|
|
cc_test(
|
|
process_mesh_test
|
|
SRCS process_mesh_test.cc
|
|
DEPS type_info)
|
|
else()
|
|
cc_test(device_mesh_test SRCS device_mesh_test.cc)
|
|
cc_test(process_mesh_test SRCS process_mesh_test.cc)
|
|
endif()
|
|
|
|
cc_test(
|
|
dist_attr_test
|
|
SRCS dist_attr_test.cc
|
|
DEPS proto_desc)
|
|
|
|
if(WITH_DISTRIBUTE)
|
|
cc_library(
|
|
spmd_rule_test_util
|
|
SRCS spmd_rule_test_util.cc
|
|
DEPS gtest)
|
|
cc_test(
|
|
dist_tensor_test
|
|
SRCS dist_tensor_test.cc
|
|
DEPS phi common)
|
|
|
|
paddle_test(spmd_rule_test SRCS spmd_rule_test.cc DEPS spmd_rule_test_util
|
|
phi)
|
|
|
|
paddle_test(softmax_grad_spmd_rule_test SRCS softmax_grad_spmd_rule_test.cc
|
|
DEPS spmd_rule_test_util phi)
|
|
|
|
paddle_test(tile_spmd_rule_test SRCS tile_spmd_rule_test.cc DEPS
|
|
spmd_rule_test_util phi)
|
|
paddle_test(tile_co_shard_spmd_rule_test SRCS tile_co_shard_spmd_rule_test.cc
|
|
DEPS spmd_rule_test_util phi)
|
|
|
|
paddle_test(
|
|
fused_linear_param_grad_add_spmd_rule_test SRCS
|
|
fused_linear_param_grad_add_spmd_rule_test.cc DEPS spmd_rule_test_util phi)
|
|
|
|
paddle_test(
|
|
cross_entropy_softmax_spmd_rule_test SRCS
|
|
cross_entropy_softmax_spmd_rule_test.cc DEPS spmd_rule_test_util phi)
|
|
|
|
paddle_test(expand_spmd_rule_test SRCS expand_spmd_rule_test.cc DEPS
|
|
spmd_rule_test_util phi)
|
|
|
|
paddle_test(expand_as_spmd_rule_test SRCS expand_as_spmd_rule_test.cc DEPS
|
|
spmd_rule_test_util phi)
|
|
|
|
paddle_test(matmul_co_shard_spmd_rule_test SRCS
|
|
matmul_co_shard_spmd_rule_test.cc DEPS spmd_rule_test_util phi)
|
|
|
|
paddle_test(custom_op_spmd_rule_test SRCS custom_op_spmd_rule_test.cc DEPS
|
|
spmd_rule_test_util phi)
|
|
|
|
paddle_test(fused_rms_norm_spmd_rule_test SRCS
|
|
fused_rms_norm_spmd_rule_test.cc DEPS spmd_rule_test_util phi)
|
|
paddle_test(moe_gate_dispatch_spmd_rule_test SRCS
|
|
moe_gate_dispatch_spmd_rule_test.cc DEPS spmd_rule_test_util phi)
|
|
|
|
paddle_test(moe_combine_spmd_rule_test SRCS moe_combine_spmd_rule_test.cc
|
|
DEPS spmd_rule_test_util phi)
|
|
|
|
paddle_test(softmax_co_shard_spmd_rule_test SRCS
|
|
softmax_co_shard_spmd_rule_test.cc DEPS spmd_rule_test_util phi)
|
|
|
|
paddle_test(
|
|
index_select_co_shard_spmd_rule_test SRCS
|
|
index_select_co_shard_spmd_rule_test.cc DEPS spmd_rule_test_util phi)
|
|
|
|
paddle_test(reshape_co_shard_spmd_rule_test SRCS
|
|
reshape_co_shard_spmd_rule_test.cc DEPS spmd_rule_test_util phi)
|
|
|
|
paddle_test(argsort_co_shard_spmd_rule_test SRCS
|
|
argsort_co_shard_spmd_rule_test.cc DEPS spmd_rule_test_util phi)
|
|
paddle_test(transpose_co_shard_spmd_rule_test SRCS
|
|
transpose_co_shard_spmd_rule_test.cc DEPS spmd_rule_test_util phi)
|
|
|
|
endif()
|
|
|
|
if(WIN32)
|
|
cc_test(
|
|
dist_mapper_test
|
|
SRCS dist_mapper_test.cc
|
|
DEPS type_info)
|
|
else()
|
|
cc_test(
|
|
dist_mapper_test
|
|
SRCS dist_mapper_test.cc
|
|
DEPS phi)
|
|
endif()
|