Fix _get_data_into_warp.

PiperOrigin-RevId: 817681130 Change-Id: Iceb622e5655021ac25388ec28f1f78fe61f58ddc
2025-10-10 09:43:27 -07:00
parent 624ae0ac26
commit 5f55e96810
1 changed files with 1 additions and 1 deletions
@@ -1264,7 +1264,7 @@ def _get_data_into_warp(
        else d
    )
    result_i = result[i] if batched else result
-    ncon = d_i._impl.ncon[0]
+    ncon = d_i._impl.nacon[0]
    nefc = int(d_i._impl.nefc)
    # nj = int(d_i._impl.nj[0])
    nj = 0  # TODO(btaba): add nj back