We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
accelerator
INC_TARGET_DEVICE
1 parent c9e265f commit 68becc4Copy full SHA for 68becc4
neural_compressor/torch/utils/environ.py
@@ -150,6 +150,10 @@ def get_accelerator(device_name="auto"):
150
from neural_compressor.torch.utils.auto_accelerator import auto_detect_accelerator
151
152
accelerator = auto_detect_accelerator(device_name)
153
+ inc_target_device = os.environ.get("INC_TARGET_DEVICE", None)
154
+ if inc_target_device is not None and accelerator._name != inc_target_device.lower():
155
+ auto_detect_accelerator.cache_clear()
156
+ accelerator = auto_detect_accelerator(device_name)
157
return accelerator
158
159
0 commit comments