* add reduction spmd rule for auto parallel

* fix the logic of handling partial

* fix code style

* fix the partial handling