add benchmarks for deep_clone

PiperOrigin-RevId: 856181788
Change-Id: I5d91d4fa88462c8d93fd493ecf4f5381f20a6833

Author: Koladata Team

koladata/internal/op_utils/BUILD

@@ -395,6 +395,7 @@ cc_test(
     srcs = ["deep_clone_test.cc"],
     deps = [
         ":deep_clone",
+        ":presence_and",
         "//koladata:test_utils",
         "//koladata/internal:data_bag",
         "//koladata/internal:data_item",
@@ -405,8 +406,10 @@ cc_test(
         "//koladata/internal:uuid_object",
         "//koladata/internal/testing:deep_op_utils",
         "//koladata/internal/testing:matchers",
+        "@com_google_absl//absl/random",
         "@com_google_absl//absl/types:span",
         "@com_google_arolla//arolla/dense_array",
+        "@com_google_arolla//arolla/memory",
         "@com_google_arolla//arolla/qtype",
         "@com_google_arolla//arolla/util",
         "@com_google_googletest//:gtest_main",
@@ -729,6 +732,23 @@ cc_library(
     ],
 )
 
+cc_test(
+    name = "deep_clone_benchmarks",
+    srcs = ["deep_clone_benchmarks.cc"],
+    deps = [
+        ":deep_clone",
+        ":deep_op_benchmarks_util",
+        "//koladata/internal:data_bag",
+        "//koladata/internal:data_item",
+        "//koladata/internal:data_slice",
+        "@com_google_absl//absl/log:check",
+        "@com_google_arolla//arolla/qtype",
+        "@com_google_arolla//arolla/util/testing",
+        "@com_google_benchmark//:benchmark_main",
+        "@com_google_googletest//:gtest",
+    ],
+)
+
 cc_test(
     name = "extract_benchmarks",
     srcs = ["extract_benchmarks.cc"],

koladata/internal/op_utils/deep_clone_benchmarks.cc

@@ -0,0 +1,64 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+#include "benchmark/benchmark.h"
+#include "absl/log/check.h"
+#include "arolla/qtype/base_types.h"
+#include "koladata/internal/data_bag.h"
+#include "koladata/internal/data_item.h"
+#include "koladata/internal/data_slice.h"
+#include "koladata/internal/op_utils/deep_op_benchmarks_util.h"
+#include "koladata/internal/op_utils/deep_clone.h"
+
+namespace koladata::internal {
+namespace {
+
+using benchmarks_utils::RunBenchmarksFn;
+
+void RunBenchmarks(benchmark::State& state, DataSliceImpl& ds, DataItem& schema,
+                   DataBagImplPtr& databag,
+                   DataBagImpl::FallbackSpan fallbacks) {
+  while (state.KeepRunning()) {
+    benchmark::DoNotOptimize(ds);
+    benchmark::DoNotOptimize(schema);
+    benchmark::DoNotOptimize(databag);
+    benchmark::DoNotOptimize(fallbacks);
+    auto result_db = DataBagImpl::CreateEmptyDatabag();
+    DeepCloneOp(result_db.get())(ds, schema, *databag, fallbacks).IgnoreError();
+    benchmark::DoNotOptimize(result_db);
+  }
+}
+
+void BM_DisjointChains(benchmark::State& state) {
+  benchmarks_utils::BM_DisjointChains(state, RunBenchmarks);
+}
+BENCHMARK(BM_DisjointChains)->Apply(benchmarks_utils::kBenchmarkFn);
+
+void BM_DisjointChainsObjects(benchmark::State& state) {
+  benchmarks_utils::BM_DisjointChainsObjects(state, RunBenchmarks);
+}
+BENCHMARK(BM_DisjointChainsObjects)->Apply(benchmarks_utils::kBenchmarkFn);
+
+void BM_DAG(benchmark::State& state) {
+  benchmarks_utils::BM_DAG(state, RunBenchmarks);
+}
+BENCHMARK(BM_DAG)->Apply(benchmarks_utils::kLayersBenchmarkFn);
+
+void BM_DAGObjects(benchmark::State& state) {
+  benchmarks_utils::BM_DAGObjects(state, RunBenchmarks);
+}
+BENCHMARK(BM_DAGObjects)->Apply(benchmarks_utils::kLayersBenchmarkFn);
+
+}  // namespace
+}  // namespace koladata::internal