File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -60,7 +60,7 @@ def main():
6060 place = 'gpu'
6161 elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
6262 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' ):
6464 place = 'npu'
6565 elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
6666 place = 'mlu'
Original file line number Diff line number Diff line change @@ -147,7 +147,7 @@ def main(args):
147147 place = 'gpu'
148148 elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
149149 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' ):
151151 place = 'npu'
152152 else :
153153 place = 'cpu'
Original file line number Diff line number Diff line change @@ -170,7 +170,7 @@ def main(args):
170170 place = 'gpu'
171171 elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
172172 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' ):
174174 place = 'npu'
175175 elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
176176 place = 'mlu'
Original file line number Diff line number Diff line change @@ -141,7 +141,7 @@ def main(args):
141141 place = 'gpu'
142142 elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
143143 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' ):
145145 place = 'npu'
146146 else :
147147 place = 'cpu'
Original file line number Diff line number Diff line change @@ -155,7 +155,7 @@ def main(args):
155155 place = 'gpu'
156156 elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
157157 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' ):
159159 place = 'npu'
160160 elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
161161 place = 'mlu'
Original file line number Diff line number Diff line change @@ -175,7 +175,7 @@ def main(args):
175175 place = 'gpu'
176176 elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
177177 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' ):
179179 place = 'npu'
180180 elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
181181 place = 'mlu'
Original file line number Diff line number Diff line change @@ -149,7 +149,7 @@ def main(args):
149149 place = 'gpu'
150150 elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
151151 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' ):
153153 place = 'npu'
154154 elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
155155 place = 'mlu'
Original file line number Diff line number Diff line change @@ -148,7 +148,7 @@ def main(args):
148148 place = 'gpu'
149149 elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
150150 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' ):
152152 place = 'npu'
153153 elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
154154 place = 'mlu'
Original file line number Diff line number Diff line change @@ -168,7 +168,7 @@ def main(args):
168168 place = 'gpu'
169169 elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
170170 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' ):
172172 place = 'npu'
173173 elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
174174 place = 'mlu'
Original file line number Diff line number Diff line change @@ -142,7 +142,7 @@ def main(args):
142142 place = 'gpu'
143143 elif args .device == 'xpu' and paddle .is_compiled_with_xpu ():
144144 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' ):
146146 place = 'npu'
147147 elif args .device == 'mlu' and paddle .is_compiled_with_mlu ():
148148 place = 'mlu'
You can’t perform that action at this time.
0 commit comments