Skip to content

Commit

Permalink
Automated Code Change
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 674603528
  • Loading branch information
Google-ML-Automation committed Sep 14, 2024
1 parent 91e71d2 commit ad171d8
Show file tree
Hide file tree
Showing 5 changed files with 3 additions and 7 deletions.
2 changes: 1 addition & 1 deletion xla/service/hlo_creation_utils.cc
Original file line number Diff line number Diff line change
Expand Up @@ -796,7 +796,7 @@ HloInstruction* CreateDummyOp(HloComputation::Builder* b, const Shape& shape) {
absl::StatusOr<std::unique_ptr<HloComputation>> CreateComputationWithSignature(
absl::Span<const Shape* const> domain, const Shape& range,
absl::string_view name) {
HloComputation::Builder b{std::string(name)};
HloComputation::Builder b{name};
int64_t param_idx = 0;
for (const Shape* param_shape : domain) {
b.AddInstruction(HloInstruction::CreateParameter(
Expand Down
2 changes: 1 addition & 1 deletion xla/service/hlo_dataflow_analysis_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -2106,7 +2106,7 @@ TEST_F(HloDataflowAnalysisTest, AllGatherStartAndDoneWithTuple) {
TF_ASSERT_OK_AND_ASSIGN(module_, ParseAndReturnVerifiedModule(hlo_text));
const HloDataflowAnalysis& analysis = RunAnalysis(/*ssa_form=*/false);
absl::Status status = analysis.Verify();
EXPECT_TRUE(status.ok()) << status.ToString();
EXPECT_TRUE(status.ok()) << status;

HloInstruction* done = module_->entry_computation()->root_instruction();
HloInstruction* start = done->mutable_operand(0);
Expand Down
2 changes: 0 additions & 2 deletions xla/service/host_offload_legalize_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -37,8 +37,6 @@ limitations under the License.
#include "xla/util.h"
#include "tsl/platform/statusor.h"

namespace m = ::xla::match;

namespace xla {
namespace {

Expand Down
2 changes: 0 additions & 2 deletions xla/service/host_offloader.cc
Original file line number Diff line number Diff line change
Expand Up @@ -63,8 +63,6 @@ namespace xla {

namespace {

using ::xla::host_memory_offload_annotations::kMoveToDeviceCustomCallTarget;
using ::xla::host_memory_offload_annotations::kMoveToHostCustomCallTarget;
using ::xla::host_offload_utils::InstructionAndShapeIndex;

void SetMemorySpace(Shape* shape, int64_t memory_space_color) {
Expand Down
2 changes: 1 addition & 1 deletion xla/service/layout_assignment.cc
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,7 @@ BufferLayoutConstraint::BufferLayoutConstraint(const Layout& layout,
bool mandatory, bool dfs,
int64_t priority)
: LayoutConstraint(mandatory, dfs, priority), buffer_(&buffer) {
CHECK(LayoutUtil::ValidateLayoutForShape(layout, buffer.shape()).ok());
CHECK_OK(LayoutUtil::ValidateLayoutForShape(layout, buffer.shape()));
layout_.push_back(layout);
}

Expand Down

0 comments on commit ad171d8

Please sign in to comment.