From 443b920ab4ff58cc73093c229df3cc9980d89365 Mon Sep 17 00:00:00 2001 From: jiaqiw Date: Thu, 19 Oct 2023 09:38:32 +0800 Subject: [PATCH] Fix typo 'torch.npu.set_device("cuda:0")' of apis doc --- docs/api/torch_npu_apis.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/api/torch_npu_apis.md b/docs/api/torch_npu_apis.md index 98560562ea..d99aff6acb 100644 --- a/docs/api/torch_npu_apis.md +++ b/docs/api/torch_npu_apis.md @@ -9528,8 +9528,8 @@ Take torch.cuda.set_device as an example. ```Python ... -torch.cuda.set_device("cuda:0") -# torch.npu.set_device("cuda:0") +# torch.cuda.set_device("cuda:0") +torch.npu.set_device("npu:0") ... ``` -- Gitee