torch_npu.utils.reset_thread_affinity

产品支持情况

产品	是否支持
Atlas A3 训练系列产品	√
Atlas A3 推理系列产品	√
Atlas A2 训练系列产品	√
Atlas A2 推理系列产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√

功能说明

恢复当前线程的绑核区间为主线程。

函数原型

torch_npu.utils.reset_thread_affinity()

参数说明

无

返回值说明

无

约束说明

该接口需要环境变量CPU_AFFINITY_CONF的mode设置为1或2时才生效，一般在拉起子线程的位置后使用，恢复当前线程的绑核区间为主线程区间。推荐和torch_npu.utils.set_thread_affinity配套使用。

调用示例

>>> import torch_npu
>>> import threading
>>>
>>> def run_thread():
...   print("This is a child thread.")
>>>
>>> torch_npu.utils.set_thread_affinity([12, 19])
>>> child_thread = threading.Thread(target=run_thread)
>>> child_thread.start()
>>> torch_npu.utils.reset_thread_affinity()
>>> child_thread.join()