Enve's fix for forward_cpu