Fix #2508.
PiperOrigin-RevId: 814761747 Change-Id: Ic3d1ed359f377799126987c3640b7e534e764c6b
This commit is contained in:
committed by
Copybara-Service
parent
d321ba7642
commit
6ae9cc8061
@@ -1466,7 +1466,7 @@ def _get_data_into(
|
||||
if d.impl == types.Impl.JAX:
|
||||
if field.name == 'qM' and not support.is_sparse(m):
|
||||
value = value[dof_i, dof_j]
|
||||
elif field.name == 'qLD' and not support.is_sparse(m):
|
||||
elif field.name == 'qLD':
|
||||
value = np.zeros(m.nC)
|
||||
elif field.name == 'qLDiagInv' and not support.is_sparse(m):
|
||||
value = np.ones(m.nv)
|
||||
|
||||
Reference in New Issue
Block a user