Skip to content

Commit 2f60bed

Browse files
authored
[fix][npu] update npu api (#3151)
1 parent d676bcf commit 2f60bed

File tree

10 files changed

+10
-10
lines changed

10 files changed

+10
-10
lines changed

contrib/QualityInspector/tools/end2end/predict.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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'

contrib/QualityInspector/tools/seg/predict.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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'

contrib/QualityInspector/tools/seg/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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'

contrib/QualityInspector/tools/seg/val.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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'

contrib/RSSegBenchmark/c2fnet/predict.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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'

contrib/RSSegBenchmark/c2fnet/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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'

contrib/RSSegBenchmark/c2fnet/val.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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'

contrib/RSSegBenchmark/predict.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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'

contrib/RSSegBenchmark/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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'

contrib/RSSegBenchmark/val.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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'

0 commit comments

Comments
 (0)