Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions paddle/fluid/pir/dialect/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -264,6 +264,9 @@ file(GLOB_RECURSE dist_dialect_srcs
set(op_dialect_srcs ${op_dialect_srcs} ${dist_dialect_srcs})
# endif()
set(op_dialect_deps phi common pir type_info string_helper)
if(WITH_ROCM)
set(op_dialect_deps ${op_dialect_deps} global_utils)
endif()

cc_library(
op_dialect
Expand Down
14 changes: 8 additions & 6 deletions test/cpp/auto_parallel/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -14,20 +14,22 @@ if(WITH_DISTRIBUTE)
SRCS dist_tensor_test.cc
DEPS phi common)

paddle_test(spmd_rule_test SRCS spmd_rule_test.cc DEPS spmd_rule_test_util)
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)
DEPS spmd_rule_test_util phi)

paddle_test(tile_spmd_rule_test SRCS tile_spmd_rule_test.cc DEPS
spmd_rule_test_util)
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)
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)
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_as_spmd_rule_test SRCS expand_as_spmd_rule_test.cc DEPS
spmd_rule_test_util phi)
Expand Down