Skip to content

Commit bb09ae7

Browse files
[NPU] fix ut question: contiguous. (#1366)
1 parent eca5ecc commit bb09ae7

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

backends/npu/tests/unittests/test_contiguous_op_npu.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,17 @@
1414

1515
from __future__ import print_function
1616

17+
import os
18+
19+
os.environ["FLAGS_use_stride_kernel"] = "1"
20+
1721
import paddle
1822
import unittest
1923
import numpy as np
2024
from npu_utils import check_soc_version
2125

26+
paddle.set_flags({"FLAGS_use_stride_kernel": 1})
27+
2228

2329
class TestContiguous(unittest.TestCase):
2430
def setUp(self):
Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
disable_ut_npu
22
test_check_nan_inf_op_npu
33
test_conv3d_op_npu
4-
test_contiguous_op_npu
54
test_fused_matmul_bias_op_npu
65
test_zero_dim_tensor_npu
76
test_matmulv2_op_npu

0 commit comments

Comments
 (0)