Skip to content

Commit fa7bb22

Browse files
fix typo
1 parent 4d16274 commit fa7bb22

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

deepmd/pd/utils/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,7 @@ def get_generator(
349349
elif DEVICE == "xpu":
350350
generator = paddle.framework.core.default_xpu_generator(0)
351351
elif DEVICE.startswith("xpu:"):
352-
generator = paddle.framework.core.default_cuda_generator(
352+
generator = paddle.framework.core.default_xpu_generator(
353353
int(DEVICE.split("xpu:")[1])
354354
)
355355
else:

0 commit comments

Comments
 (0)