conjugate_gradient: Fix for multi-D non-Field input

26 jobs for misc_edh in 6 seconds (queued for 4 seconds)