From 16f6eae54fa59af57b02358e587e3760cdd727a1 Mon Sep 17 00:00:00 2001 From: ralberd Date: Fri, 29 Aug 2025 13:47:59 -0600 Subject: [PATCH] minor fixes for multi block with properties --- optimism/Mechanics.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/optimism/Mechanics.py b/optimism/Mechanics.py index ba605314..7b11f07f 100644 --- a/optimism/Mechanics.py +++ b/optimism/Mechanics.py @@ -257,9 +257,10 @@ def _compute_element_stiffnesses_multi_block(U, stateVariables, props, dt, funct materialModel = blockModels[blockKey] blockProps = props[blockKey] L = strain_energy_density_to_lagrangian_density(materialModel.compute_energy_density) + vmapValue = vmapPropValue(blockProps) elemIds = functionSpace.mesh.blocks[blockKey] f = vmap(compute_element_stiffness_from_global_fields, - (None, None, 0, None, 0, 0, 0, 0, None, None)) + (None, None, 0, vmapValue, None, 0, 0, 0, 0, None, None)) blockHessians = f(U, fs.mesh.coords, stateVariables[elemIds], blockProps, dt, fs.mesh.conns[elemIds], fs.shapes[elemIds], fs.shapeGrads[elemIds], fs.vols[elemIds], L, modify_element_gradient)