* [Feat] Support mps

* fix docstring

* support mlu

* add ut and refine docstring

* move get_max_cuda_memory and set_multi_processing to a public function

* fix lint

* fix lint

* fix lint

* delete _set_multi_processing

* fix error

* rename