accelerate
8576112b - enable 2 UT cases on XPU (#3445)

Commit
318 days ago
enable 2 UT cases on XPU (#3445) * enable test_dispatch_model_tied_weights_memory_with_nested_offload_cpu test case on XPU Signed-off-by: root <root@a4bf01945cfe.jf.intel.com> * fix style Signed-off-by: Yao, Matrix <matrix.yao@intel.com> * enable test_dispatch_model_tied_weights_memory on XPU Signed-off-by: N <matrix.yao@intel.com> * fix bug Signed-off-by: root <root@a4bf01945cfe.jf.intel.com> * Update src/accelerate/test_utils/testing.py Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com> * Update src/accelerate/test_utils/testing.py Co-authored-by: Zach Mueller <muellerzr@gmail.com> * Update tests/test_big_modeling.py Co-authored-by: Zach Mueller <muellerzr@gmail.com> * fix style Signed-off-by: Yao, Matrix <matrix.yao@intel.com> --------- Signed-off-by: root <root@a4bf01945cfe.jf.intel.com> Signed-off-by: Yao, Matrix <matrix.yao@intel.com> Signed-off-by: N <matrix.yao@intel.com> Co-authored-by: root <root@a4bf01945cfe.jf.intel.com> Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com> Co-authored-by: Zach Mueller <muellerzr@gmail.com>
Author
Parents
Loading