We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7a72522 commit 3ba4aa0Copy full SHA for 3ba4aa0
python/paddle/incubate/tensor/manipulation.py
@@ -108,7 +108,10 @@ def create_async_load():
108
Constructs a new AsyncLoad object.
109
It is used to load/reload data asynchronously on GPU.
110
"""
111
- return core.AsyncLoad()
+ if paddle.is_compiled_with_xpu():
112
+ return core.XpuAsyncLoad()
113
+ else: # default is GPU or CUDA
114
+ return core.AsyncLoad()
115
116
117
def create_xpu_async_load():
0 commit comments