* support set master_grad

* move register_hook to auto_cast

* update unittest

* fix fp16 test

* update for review comments