2025-01-25 nightly release (526902f)

pytorch · Jan 25, 2025 · d245db3 · d245db3
1 parent 72845b4
commit d245db3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchrec/metrics/gauc.py b/torchrec/metrics/gauc.py
@@ -100,6 +100,7 @@ def to_3d(
     return torch.ops.fbgemm.jagged_2d_to_dense(tensor_2d, offsets, max_length)
 
 
+@torch.compiler.disable
 def get_auc_states(
     labels: torch.Tensor,
     predictions: torch.Tensor,