From e6ec0e95eb6e211b05de45e2500e15dc8e4a81e1 Mon Sep 17 00:00:00 2001 From: Jennifer Lee Date: Fri, 29 May 2026 14:46:01 -0700 Subject: [PATCH] Add measurements framework (#789) Summary: Add a measurements framework that evaluates different compression methods by running multiple strategies (ML selector, best-successor per cluster, brute-force per sample, generic ML, plain zstd) on captured sample data and printing a side-by-side comparison table. Added test cases for the measurements and python integration so managed compression trainer can trigger measurements if --capture-data is specified Differential Revision: D106423347 --- tools/ml_selector/BUCK | 4 ++++ tools/training/ace/BUCK | 4 ++++ 2 files changed, 8 insertions(+) diff --git a/tools/ml_selector/BUCK b/tools/ml_selector/BUCK index 273a04e09..f60155187 100644 --- a/tools/ml_selector/BUCK +++ b/tools/ml_selector/BUCK @@ -11,6 +11,10 @@ zs_cxxlibrary( compiler_flags = [ "-Wno-unused-exception-parameter", # For XGBoost warnings ], + visibility = [ + "fbcode//common/managed_compression/...", + "//openzl/...", + ], deps = [ "../..:zstronglib", "../../cpp:openzl_cpp", diff --git a/tools/training/ace/BUCK b/tools/training/ace/BUCK index 7eccf4273..653655091 100644 --- a/tools/training/ace/BUCK +++ b/tools/training/ace/BUCK @@ -30,6 +30,10 @@ cpp_library( "crowding_distance_selector.h", ]), header_namespace = "", + visibility = [ + "fbcode//common/managed_compression/...", + "//openzl/...", + ], deps = [ "../../..:zstronglib", ],