* regist pytorch ddp and dp, add unit test

* minor refine

* Support check custom wrapper

* enhance ut