cc_test(device_mesh_test SRCS device_mesh_test.cc)

cc_test(process_mesh_test SRCS process_mesh_test.cc)

cc_test(dist_attr_test SRCS dist_attr_test.cc)

if(WITH_DISTRIBUTE)
  cc_test(
    dist_tensor_test
    SRCS dist_tensor_test.cc
    DEPS phi)
endif()

cc_test_old(dist_mapper_test SRCS dist_mapper_test.cc DEPS phi)

cc_test_old(spmd_rule_test SRCS spmd_rule_test.cc DEPS spmd_rules)