Update logic for enabling grad_on_execution in lightning_*.py devices

Update logic for enabling grad_on_execution in lightning_*.py devices #10515