Fix _get_data_into_warp.
PiperOrigin-RevId: 817681130 Change-Id: Iceb622e5655021ac25388ec28f1f78fe61f58ddc
This commit is contained in:
committed by
Copybara-Service
parent
624ae0ac26
commit
5f55e96810
@@ -1264,7 +1264,7 @@ def _get_data_into_warp(
|
|||||||
else d
|
else d
|
||||||
)
|
)
|
||||||
result_i = result[i] if batched else result
|
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)
|
nefc = int(d_i._impl.nefc)
|
||||||
# nj = int(d_i._impl.nj[0])
|
# nj = int(d_i._impl.nj[0])
|
||||||
nj = 0 # TODO(btaba): add nj back
|
nj = 0 # TODO(btaba): add nj back
|
||||||
|
|||||||
Reference in New Issue
Block a user