SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Enhance IPEX integration in Trainer (#18072)

Browse Source 
* enhance ipex import

* refine codes

* refine style

* add link

* style

Co-authored-by: Stas Bekman <stas@stason.org>
This commit is contained in:
jianan-gu
2022-07-12 12:34:09 +08:00
committed by GitHub
parent a462fc9232
commit b7d8bd378c
4 changed files with 31 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/transformers/testing_utils.py
View File

@@ -292,10 +292,15 @@ def require_intel_extension_for_pytorch(test_case):
"""
Decorator marking a test that requires Intel Extension for PyTorch.
These tests are skipped when Intel Extension for PyTorch isn't installed.
These tests are skipped when Intel Extension for PyTorch isn't installed or it does not match current PyTorch
version.
"""
return unittest.skipUnless(is_ipex_available(), "test requires Intel Extension for PyTorch")(test_case)
return unittest.skipUnless(
is_ipex_available(),
"test requires Intel Extension for PyTorch to be installed and match current PyTorch version, see"
" https://github.com/intel/intel-extension-for-pytorch",
)(test_case)
def require_torch_scatter(test_case):