* fix fp16 dtype checking for argmax op

* run fp16 test when place is gpu

* Update search.py

fix doc