Skip to content

Commit e104e58

Browse files
committed
fix device
1 parent e74017c commit e104e58

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/paddle/fluid/tests/unittests/npu/test_merged_momentum_op_npu.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -261,7 +261,7 @@ def setUp(self):
261261
paddle.enable_static()
262262
self.shapes = [[3, 4], [2, 7], [5, 6], [7, 8]]
263263
self.seed = 10
264-
self.place = paddle.device.NPUPlace(0)
264+
self.place = paddle.fluid.NPUPlace(0)
265265
self.__class__.use_npu = True
266266

267267
def gen_rand_data(self, shapes, dtype):
@@ -314,7 +314,7 @@ def setUp(self):
314314
paddle.enable_static()
315315
self.shapes = [[3, 4], [2, 7], [5, 6], [7, 8]]
316316
self.seed = 10
317-
self.place = paddle.device.NPUPlace(0)
317+
self.place = paddle.fluid.NPUPlace(0)
318318
self.__class__.use_npu = True
319319

320320
def gen_rand_data(self, shapes, dtype):

0 commit comments

Comments
 (0)