sglang/python/sglang/srt/managers/tp_worker_overlap_thread.py

```
    def forward_thread_func(self):
        try:
            current_device = torch.npu.current_device()
            torch.npu.set_device(current_device)
```