Skip to content

Commit c203958

Browse files
ariellubonjaclaude
andcommitted
[YDF] Faster OOB computation
Cherry-pick from google#324. Replaces single mutex OOB computation with striped locking and an OOB gate mechanism for better multi-threaded performance. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent 6d9a7d1 commit c203958

4 files changed

Lines changed: 297 additions & 126 deletions

File tree

yggdrasil_decision_forests/learner/random_forest/BUILD

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,13 +55,15 @@ cc_library_ydf(
5555
"//yggdrasil_decision_forests/utils:synchronization_primitives",
5656
"//yggdrasil_decision_forests/utils:time",
5757
"//yggdrasil_decision_forests/utils:parallel_chrono",
58+
"@com_google_absl//absl/cleanup",
5859
"@com_google_absl//absl/container:flat_hash_set",
5960
"@com_google_absl//absl/log",
6061
"@com_google_absl//absl/status",
6162
"@com_google_absl//absl/status:statusor",
6263
"@com_google_absl//absl/strings",
6364
"@com_google_absl//absl/strings:str_format",
6465
"@com_google_absl//absl/time",
66+
"@com_google_absl//absl/types:span",
6567
],
6668
alwayslink = 1,
6769
)
@@ -126,6 +128,7 @@ cc_test(
126128
"@com_google_absl//absl/status:statusor",
127129
"@com_google_absl//absl/strings",
128130
"@com_google_absl//absl/time",
131+
"@com_google_absl//absl/types:span",
129132
"@com_google_googletest//:gtest_main",
130133
],
131134
)

0 commit comments

Comments
 (0)