DefaultXlaOptions().device_type);

  MockXlaCompilerClient mock_client;
  EXPECT_CALL(mock_client, SerializeExecutable(_))
      .WillOnce(Return(errors::Unimplemented("Unimplemented.")));
  EXPECT_CALL(mock_client, BuildSerializedExecutable(_, _))
      .WillOnce(Return(serialized_xla_executable_));

  // the compilation finshes to verify expected behavior.
  Notification done;
  EXPECT_CALL(*mock_profiler_,
              ShouldCompileCluster(_, DeviceCompileMode::kAsync, 1))
      .WillOnce(Return(true));
  EXPECT_CALL(*mock_profiler_, RegisterCompilation(_, _, false))
      .WillOnce([&done] {
        done.Notify();
        return absl::OkStatus();
      });

  ASSERT_THAT(remat.GetMemProfile(),
              ElementsAreArray({1, 3, 7, 15, 23, 19, 9, 4}));

  EXPECT_CALL(remat, ApplyRemat(FieldsAre(/*begin=*/2,
                                          /*end=*/3,
                                          /*insert=*/5)));
  EXPECT_CALL(remat, ApplyRemat(FieldsAre(/*begin=*/0,
                                          /*end=*/1,