Fix all is_torch_tpu_available issues (#17936)

* Fix all is_torch_tpu_available
2022-06-29 11:03:33 -04:00
parent 77b76672e2
commit 7c4c6f6084
10 changed files with 22 additions and 19 deletions
--- a/examples/research_projects/quantization-qdqbert/trainer_quant_qa.py
+++ b/examples/research_projects/quantization-qdqbert/trainer_quant_qa.py
@@ -30,7 +30,7 @@ from transformers.trainer_utils import PredictionOutput

 logger = logging.getLogger(__name__)

-if is_torch_tpu_available():
+if is_torch_tpu_available(check_device=False):
    import torch_xla.core.xla_model as xm
    import torch_xla.debug.metrics as met