We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4d16274 commit fa7bb22Copy full SHA for fa7bb22
1 file changed
deepmd/pd/utils/utils.py
@@ -349,7 +349,7 @@ def get_generator(
349
elif DEVICE == "xpu":
350
generator = paddle.framework.core.default_xpu_generator(0)
351
elif DEVICE.startswith("xpu:"):
352
- generator = paddle.framework.core.default_cuda_generator(
+ generator = paddle.framework.core.default_xpu_generator(
353
int(DEVICE.split("xpu:")[1])
354
)
355
else:
0 commit comments