File tree Expand file tree Collapse file tree 10 files changed +10
-10
lines changed Expand file tree Collapse file tree 10 files changed +10
-10
lines changed Original file line number Diff line number Diff line change @@ -60,7 +60,7 @@ def main():
60
60
place = 'gpu'
61
61
elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
62
62
place = 'xpu'
63
- elif args .device == 'npu' and paddle .is_compiled_with_npu ( ):
63
+ elif args .device == 'npu' and paddle .is_compiled_with_custom_device ( 'npu' ):
64
64
place = 'npu'
65
65
elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
66
66
place = 'mlu'
Original file line number Diff line number Diff line change @@ -147,7 +147,7 @@ def main(args):
147
147
place = 'gpu'
148
148
elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
149
149
place = 'xpu'
150
- elif args .device == 'npu' and paddle .is_compiled_with_npu ( ):
150
+ elif args .device == 'npu' and paddle .is_compiled_with_custom_device ( 'npu' ):
151
151
place = 'npu'
152
152
else :
153
153
place = 'cpu'
Original file line number Diff line number Diff line change @@ -170,7 +170,7 @@ def main(args):
170
170
place = 'gpu'
171
171
elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
172
172
place = 'xpu'
173
- elif args .device == 'npu' and paddle .is_compiled_with_npu ( ):
173
+ elif args .device == 'npu' and paddle .is_compiled_with_custom_device ( 'npu' ):
174
174
place = 'npu'
175
175
elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
176
176
place = 'mlu'
Original file line number Diff line number Diff line change @@ -141,7 +141,7 @@ def main(args):
141
141
place = 'gpu'
142
142
elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
143
143
place = 'xpu'
144
- elif args .device == 'npu' and paddle .is_compiled_with_npu ( ):
144
+ elif args .device == 'npu' and paddle .is_compiled_with_custom_device ( 'npu' ):
145
145
place = 'npu'
146
146
else :
147
147
place = 'cpu'
Original file line number Diff line number Diff line change @@ -155,7 +155,7 @@ def main(args):
155
155
place = 'gpu'
156
156
elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
157
157
place = 'xpu'
158
- elif args .device == 'npu' and paddle .is_compiled_with_npu ( ):
158
+ elif args .device == 'npu' and paddle .is_compiled_with_custom_device ( 'npu' ):
159
159
place = 'npu'
160
160
elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
161
161
place = 'mlu'
Original file line number Diff line number Diff line change @@ -175,7 +175,7 @@ def main(args):
175
175
place = 'gpu'
176
176
elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
177
177
place = 'xpu'
178
- elif args .device == 'npu' and paddle .is_compiled_with_npu ( ):
178
+ elif args .device == 'npu' and paddle .is_compiled_with_custom_device ( 'npu' ):
179
179
place = 'npu'
180
180
elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
181
181
place = 'mlu'
Original file line number Diff line number Diff line change @@ -149,7 +149,7 @@ def main(args):
149
149
place = 'gpu'
150
150
elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
151
151
place = 'xpu'
152
- elif args .device == 'npu' and paddle .is_compiled_with_npu ( ):
152
+ elif args .device == 'npu' and paddle .is_compiled_with_custom_device ( 'npu' ):
153
153
place = 'npu'
154
154
elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
155
155
place = 'mlu'
Original file line number Diff line number Diff line change @@ -148,7 +148,7 @@ def main(args):
148
148
place = 'gpu'
149
149
elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
150
150
place = 'xpu'
151
- elif args .device == 'npu' and paddle .is_compiled_with_npu ( ):
151
+ elif args .device == 'npu' and paddle .is_compiled_with_custom_device ( 'npu' ):
152
152
place = 'npu'
153
153
elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
154
154
place = 'mlu'
Original file line number Diff line number Diff line change @@ -168,7 +168,7 @@ def main(args):
168
168
place = 'gpu'
169
169
elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
170
170
place = 'xpu'
171
- elif args .device == 'npu' and paddle .is_compiled_with_npu ( ):
171
+ elif args .device == 'npu' and paddle .is_compiled_with_custom_device ( 'npu' ):
172
172
place = 'npu'
173
173
elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
174
174
place = 'mlu'
Original file line number Diff line number Diff line change @@ -142,7 +142,7 @@ def main(args):
142
142
place = 'gpu'
143
143
elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
144
144
place = 'xpu'
145
- elif args .device == 'npu' and paddle .is_compiled_with_npu ( ):
145
+ elif args .device == 'npu' and paddle .is_compiled_with_custom_device ( 'npu' ):
146
146
place = 'npu'
147
147
elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
148
148
place = 'mlu'
You can’t perform that action at this time.
0 commit comments