[go: nahoru, domu]

Skip to content

Commit

Permalink
[XLA:GPU][NFC] Move tile_analysis* to symbolic_tile*.
Browse files Browse the repository at this point in the history
Related code pertaining to tile propagation/analysis belongs in
symbolic_tile_analysis*.

PiperOrigin-RevId: 619171048
  • Loading branch information
bchetioui authored and tensorflower-gardener committed Mar 26, 2024
1 parent 5c52f8a commit 409ff7b
Show file tree
Hide file tree
Showing 6 changed files with 14 additions and 14 deletions.
14 changes: 7 additions & 7 deletions third_party/xla/xla/service/gpu/model/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -517,9 +517,9 @@ xla_cc_test(
)

cc_library(
name = "tile_analysis",
srcs = ["tile_analysis.cc"],
hdrs = ["tile_analysis.h"],
name = "symbolic_tile",
srcs = ["symbolic_tile.cc"],
hdrs = ["symbolic_tile.h"],
deps = [
":affine_map_printer",
":indexing_map",
Expand All @@ -534,13 +534,13 @@ cc_library(
)

xla_cc_test(
name = "tile_analysis_test",
srcs = ["tile_analysis_test.cc"],
name = "symbolic_tile_test",
srcs = ["symbolic_tile_test.cc"],
deps = [
":affine_map_printer",
":indexing_analysis",
":indexing_test_utils",
":tile_analysis",
":symbolic_tile",
"//xla/hlo/ir:hlo",
"//xla/tests:hlo_test_base",
"//xla/tests:xla_internal_test_main",
Expand All @@ -559,7 +559,7 @@ cc_library(
deps = [
":indexing_analysis",
":indexing_map",
":tile_analysis",
":symbolic_tile",
"//xla:status",
"//xla/hlo/ir:hlo",
"//xla/service:instruction_fusion",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/

#include "xla/service/gpu/model/tile_analysis.h"
#include "xla/service/gpu/model/symbolic_tile.h"

#include <cstdint>
#include <optional>
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/

#ifndef XLA_SERVICE_GPU_MODEL_TILE_ANALYSIS_H_
#define XLA_SERVICE_GPU_MODEL_TILE_ANALYSIS_H_
#ifndef XLA_SERVICE_GPU_MODEL_SYMBOLIC_TILE_H_
#define XLA_SERVICE_GPU_MODEL_SYMBOLIC_TILE_H_

#include <optional>
#include <ostream>
Expand Down Expand Up @@ -78,4 +78,4 @@ class SymbolicTile {
} // namespace gpu
} // namespace xla

#endif // XLA_SERVICE_GPU_MODEL_TILE_ANALYSIS_H_
#endif // XLA_SERVICE_GPU_MODEL_SYMBOLIC_TILE_H_
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ limitations under the License.
#include "xla/hlo/ir/hlo_opcode.h"
#include "xla/service/gpu/model/indexing_analysis.h"
#include "xla/service/gpu/model/indexing_map.h"
#include "xla/service/gpu/model/tile_analysis.h"
#include "xla/service/gpu/model/symbolic_tile.h"
#include "xla/service/instruction_fusion.h"
#include "xla/status.h"

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ limitations under the License.
#include "mlir/IR/MLIRContext.h" // from @llvm-project
#include "xla/hlo/ir/hlo_instruction.h"
#include "xla/service/gpu/model/indexing_map.h"
#include "xla/service/gpu/model/tile_analysis.h"
#include "xla/service/gpu/model/symbolic_tile.h"
#include "xla/service/instruction_fusion.h"

namespace xla {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/

#include "xla/service/gpu/model/tile_analysis.h"
#include "xla/service/gpu/model/symbolic_tile.h"

#include <optional>

Expand Down

0 comments on commit 409ff7b

Please sign in to comment.