Skip to content

Commit

Permalink
[xla-auto-sharding] Move implementation of CostGraph methods to .cc f…
Browse files Browse the repository at this point in the history
…ile.

Notes:
- Create auto_sharding_cost_graph.cc and move CostGraph implementations there.
- Make ToString() a const method.
- Make most pass-by-value arguments const.
- Add TODO to make class members private.
PiperOrigin-RevId: 615451743
  • Loading branch information
Matthew Fahrbach authored and tensorflower-gardener committed Mar 13, 2024
1 parent 8ab8fdc commit c7cc391
Show file tree
Hide file tree
Showing 3 changed files with 423 additions and 357 deletions.
1 change: 1 addition & 0 deletions third_party/xla/xla/hlo/experimental/auto_sharding/BUILD
Expand Up @@ -143,6 +143,7 @@ cc_library(

cc_library(
name = "auto_sharding_cost_graph",
srcs = ["auto_sharding_cost_graph.cc"],
hdrs = ["auto_sharding_cost_graph.h"],
compatible_with = get_compatible_with_libtpu_portable(),
deps = [
Expand Down

0 comments on commit c7cc391

Please sign in to comment.