accelerate
85901cdc - should set correct dtype to ipex optimize and use amp logic in native… (#1511)

Commit
2 years ago
should set correct dtype to ipex optimize and use amp logic in native… (#1511) * should set correct dtype to ipex optimize and use amp logic in native_amp logic in prepare_model Signed-off-by: Wang, Yi A <yi.a.wang@intel.com> * remove mix precision set in ipex, directly use it from accelerate state Signed-off-by: Wang, Yi A <yi.a.wang@intel.com> * raise import error if ipex is not valid in prepare ipex * Update src/accelerate/accelerator.py Co-authored-by: Zachary Mueller <muellerzr@gmail.com> --------- Signed-off-by: Wang, Yi A <yi.a.wang@intel.com> Co-authored-by: Zachary Mueller <muellerzr@gmail.com>
Author
Parents
Loading