Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
wmsofts
Paddle
提交
bb5dd203
P
Paddle
项目概览
wmsofts
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
bb5dd203
编写于
3月 02, 2023
作者:
J
jiangcheng
提交者:
GitHub
3月 02, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
[CINN] reopen some prim with cinn single test (#51081)
上级
ff7ce2ff
变更
8
隐藏空白更改
内联
并排
Showing
8 changed file
with
56 addition
and
56 deletion
+56
-56
python/paddle/fluid/tests/unittests/test_activation_op.py
python/paddle/fluid/tests/unittests/test_activation_op.py
+10
-5
python/paddle/fluid/tests/unittests/test_expand_v2_op.py
python/paddle/fluid/tests/unittests/test_expand_v2_op.py
+1
-1
python/paddle/fluid/tests/unittests/test_fill_any_like_op.py
python/paddle/fluid/tests/unittests/test_fill_any_like_op.py
+6
-6
python/paddle/fluid/tests/unittests/test_flatten_contiguous_range_op.py
...fluid/tests/unittests/test_flatten_contiguous_range_op.py
+7
-1
python/paddle/fluid/tests/unittests/test_full_like_op.py
python/paddle/fluid/tests/unittests/test_full_like_op.py
+2
-2
python/paddle/fluid/tests/unittests/test_reduce_op.py
python/paddle/fluid/tests/unittests/test_reduce_op.py
+21
-32
python/paddle/fluid/tests/unittests/test_slice_op.py
python/paddle/fluid/tests/unittests/test_slice_op.py
+8
-8
python/paddle/fluid/tests/unittests/test_softmax_op.py
python/paddle/fluid/tests/unittests/test_softmax_op.py
+1
-1
未找到文件。
python/paddle/fluid/tests/unittests/test_activation_op.py
浏览文件 @
bb5dd203
...
...
@@ -120,7 +120,7 @@ class TestExpPrimFp32(OpTest):
self
.
shape
=
[
12
,
17
]
def
skip_cinn
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
def
set_only_prim
(
self
):
pass
...
...
@@ -145,7 +145,7 @@ class TestExpPrimFp16(TestExpPrimFp32):
self
.
check_grad
([
'X'
],
'Out'
,
check_prim
=
True
)
def
skip_cinn
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
class
TestExpPrim_ZeroDim
(
TestExpPrimFp32
):
...
...
@@ -325,7 +325,7 @@ class TestSilu(TestActivation):
def
setUp
(
self
):
self
.
op_type
=
"silu"
self
.
prim_op_type
=
"comp"
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
self
.
python_api
=
paddle
.
nn
.
functional
.
silu
self
.
init_dtype
()
self
.
init_shape
()
...
...
@@ -349,13 +349,14 @@ class TestSilu(TestActivation):
class
TestSilu_ZeroDim
(
TestSilu
):
def
init_shape
(
self
):
self
.
shape
=
[]
self
.
enable_cinn
=
False
class
TestSiluFP16
(
TestActivation
):
def
setUp
(
self
):
self
.
op_type
=
"silu"
self
.
prim_op_type
=
"comp"
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
self
.
only_prim
=
True
self
.
python_api
=
paddle
.
nn
.
functional
.
silu
self
.
init_dtype
()
...
...
@@ -1199,7 +1200,7 @@ class TestSqrtPrimFp32(TestActivation):
self
.
inputs
=
{
'X'
:
OpTest
.
np_dtype_to_fluid_dtype
(
x
)}
self
.
outputs
=
{
'Out'
:
out
}
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
def
test_check_grad
(
self
):
if
self
.
dtype
==
np
.
float16
:
...
...
@@ -1216,11 +1217,13 @@ class TestSqrtPrimFp32(TestActivation):
class
TestSqrt_ZeroDim
(
TestSqrt
):
def
init_shape
(
self
):
self
.
shape
=
[]
self
.
enable_cinn
=
False
class
TestSqrtPrim_ZeroDim
(
TestSqrt
):
def
init_shape
(
self
):
self
.
shape
=
[]
self
.
enable_cinn
=
False
def
init_dtype
(
self
):
self
.
dtype
=
np
.
float32
...
...
@@ -1527,6 +1530,8 @@ class TestSin(TestActivation, TestParameter):
self
.
op_type
=
"sin"
self
.
init_dtype
()
self
.
init_shape
()
# prim not support now
self
.
enable_cinn
=
False
np
.
random
.
seed
(
1024
)
x
=
np
.
random
.
uniform
(
-
1
,
1
,
self
.
shape
).
astype
(
self
.
dtype
)
...
...
python/paddle/fluid/tests/unittests/test_expand_v2_op.py
浏览文件 @
bb5dd203
...
...
@@ -35,7 +35,7 @@ class TestExpandV2OpRank1(OpTest):
self
.
attrs
=
{
'shape'
:
self
.
shape
}
output
=
np
.
tile
(
self
.
inputs
[
'X'
],
self
.
expand_times
)
self
.
outputs
=
{
'Out'
:
output
}
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
def
init_data
(
self
):
self
.
ori_shape
=
[
100
]
...
...
python/paddle/fluid/tests/unittests/test_fill_any_like_op.py
浏览文件 @
bb5dd203
...
...
@@ -61,7 +61,7 @@ class TestFillAnyLikeOpFloat32(TestFillAnyLikeOp):
self
.
value
=
0.0
def
skip_cinn
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
@
unittest
.
skipIf
(
...
...
@@ -96,7 +96,7 @@ class TestFillAnyLikeOpValue1(TestFillAnyLikeOp):
self
.
value
=
1.0
def
skip_cinn
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
class
TestFillAnyLikeOpValue2
(
TestFillAnyLikeOp
):
...
...
@@ -104,7 +104,7 @@ class TestFillAnyLikeOpValue2(TestFillAnyLikeOp):
self
.
value
=
1e-10
def
skip_cinn
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
class
TestFillAnyLikeOpValue3
(
TestFillAnyLikeOp
):
...
...
@@ -112,7 +112,7 @@ class TestFillAnyLikeOpValue3(TestFillAnyLikeOp):
self
.
value
=
1e-100
def
skip_cinn
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
class
TestFillAnyLikeOpType
(
TestFillAnyLikeOp
):
...
...
@@ -136,7 +136,7 @@ class TestFillAnyLikeOpType(TestFillAnyLikeOp):
self
.
skip_cinn
()
def
skip_cinn
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
class
TestFillAnyLikeOpFloat16
(
TestFillAnyLikeOp
):
...
...
@@ -144,7 +144,7 @@ class TestFillAnyLikeOpFloat16(TestFillAnyLikeOp):
self
.
dtype
=
np
.
float16
def
skip_cinn
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
if
__name__
==
"__main__"
:
...
...
python/paddle/fluid/tests/unittests/test_flatten_contiguous_range_op.py
浏览文件 @
bb5dd203
...
...
@@ -28,15 +28,18 @@ class TestFlattenOp(OpTest):
self
.
prim_op_type
=
"comp"
self
.
start_axis
=
0
self
.
stop_axis
=
-
1
self
.
skip_cinn
()
self
.
init_test_case
()
self
.
inputs
=
{
"X"
:
np
.
random
.
random
(
self
.
in_shape
).
astype
(
"float64"
)}
self
.
init_attrs
()
self
.
enable_cinn
=
False
self
.
outputs
=
{
"Out"
:
self
.
inputs
[
"X"
].
reshape
(
self
.
new_shape
),
"XShape"
:
np
.
random
.
random
(
self
.
in_shape
).
astype
(
"float32"
),
}
def
skip_cinn
(
self
):
self
.
enable_cinn
=
True
def
test_check_output
(
self
):
self
.
check_output
(
no_check_set
=
[
"XShape"
],
check_eager
=
True
,
check_prim
=
True
...
...
@@ -135,6 +138,9 @@ class TestFlattenOp_6(TestFlattenOp):
self
.
stop_axis
=
-
1
self
.
new_shape
=
(
1
,)
def
skip_cinn
(
self
):
self
.
enable_cinn
=
False
def
init_attrs
(
self
):
self
.
attrs
=
{
"start_axis"
:
self
.
start_axis
,
...
...
python/paddle/fluid/tests/unittests/test_full_like_op.py
浏览文件 @
bb5dd203
...
...
@@ -145,7 +145,7 @@ class TestFullLikeOp2(TestFullLikeOp1):
self
.
dtype
=
np
.
float64
def
skip_cinn
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
class
TestFullLikeOp3
(
TestFullLikeOp1
):
...
...
@@ -155,7 +155,7 @@ class TestFullLikeOp3(TestFullLikeOp1):
self
.
dtype
=
np
.
int64
def
skip_cinn
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
@
unittest
.
skipIf
(
...
...
python/paddle/fluid/tests/unittests/test_reduce_op.py
浏览文件 @
bb5dd203
...
...
@@ -32,8 +32,7 @@ class TestSumOp(OpTest):
self
.
inputs
=
{
'X'
:
np
.
random
.
random
((
5
,
6
,
10
)).
astype
(
"float64"
)}
self
.
outputs
=
{
'Out'
:
self
.
inputs
[
'X'
].
sum
(
axis
=
0
)}
self
.
attrs
=
{
'dim'
:
[
0
]}
# reduce doesn't support float64 in cinn
self
.
enable_cinn
=
False
self
.
enable_cinn
=
True
def
test_check_output
(
self
):
self
.
check_output
(
check_eager
=
True
)
...
...
@@ -55,8 +54,7 @@ class TestSumOpFp32(OpTest):
'Out'
:
self
.
inputs
[
'X'
].
sum
(
axis
=
tuple
(
self
.
attrs
[
'dim'
]))
}
self
.
gradient
=
self
.
calc_gradient
()
# error occurred in cinn
self
.
enable_cinn
=
False
self
.
enable_cinn
=
True
def
test_check_output
(
self
):
self
.
check_output
(
check_eager
=
True
)
...
...
@@ -151,7 +149,7 @@ class TestSumOp_fp16_withInt(OpTest):
'Out'
:
self
.
inputs
[
'X'
].
sum
(
axis
=
tuple
(
self
.
attrs
[
'dim'
]))
}
self
.
gradient
=
self
.
calc_gradient
()
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
def
test_check_output
(
self
):
self
.
check_output
(
check_eager
=
True
)
...
...
@@ -182,7 +180,7 @@ class TestSumOp5D(OpTest):
self
.
attrs
=
{
'dim'
:
[
0
]}
self
.
outputs
=
{
'Out'
:
self
.
inputs
[
'X'
].
sum
(
axis
=
0
)}
# error occurred in cinn
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
def
test_check_output
(
self
):
self
.
check_output
(
check_eager
=
True
)
...
...
@@ -202,7 +200,7 @@ class TestSumOp6D(OpTest):
self
.
attrs
=
{
'dim'
:
[
0
]}
self
.
outputs
=
{
'Out'
:
self
.
inputs
[
'X'
].
sum
(
axis
=
0
)}
# error occurred in cinn
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
def
test_check_output
(
self
):
self
.
check_output
(
check_eager
=
True
)
...
...
@@ -678,8 +676,7 @@ class Test1DReduce(OpTest):
self
.
prim_op_type
=
"prim"
self
.
inputs
=
{
'X'
:
np
.
random
.
random
(
120
).
astype
(
"float64"
)}
self
.
outputs
=
{
'Out'
:
self
.
inputs
[
'X'
].
sum
(
axis
=
0
)}
# reduce doesn't support float64 in cinn.
self
.
enable_cinn
=
False
self
.
enable_cinn
=
True
def
test_check_output
(
self
):
self
.
check_output
()
...
...
@@ -696,8 +693,7 @@ class Test2DReduce0(Test1DReduce):
self
.
attrs
=
{
'dim'
:
[
0
]}
self
.
inputs
=
{
'X'
:
np
.
random
.
random
((
20
,
10
)).
astype
(
"float64"
)}
self
.
outputs
=
{
'Out'
:
self
.
inputs
[
'X'
].
sum
(
axis
=
0
)}
# reduce doesn't support float64 in cinn.
self
.
enable_cinn
=
False
self
.
enable_cinn
=
True
class
Test2DReduce1
(
Test1DReduce
):
...
...
@@ -710,8 +706,7 @@ class Test2DReduce1(Test1DReduce):
self
.
outputs
=
{
'Out'
:
self
.
inputs
[
'X'
].
sum
(
axis
=
tuple
(
self
.
attrs
[
'dim'
]))
}
# reduce doesn't support float64 in cinn.
self
.
enable_cinn
=
False
self
.
enable_cinn
=
True
class
Test3DReduce0
(
Test1DReduce
):
...
...
@@ -724,8 +719,7 @@ class Test3DReduce0(Test1DReduce):
self
.
outputs
=
{
'Out'
:
self
.
inputs
[
'X'
].
sum
(
axis
=
tuple
(
self
.
attrs
[
'dim'
]))
}
# reduce doesn't support float64 in cinn.
self
.
enable_cinn
=
False
self
.
enable_cinn
=
True
class
Test3DReduce1
(
Test1DReduce
):
...
...
@@ -738,8 +732,7 @@ class Test3DReduce1(Test1DReduce):
self
.
outputs
=
{
'Out'
:
self
.
inputs
[
'X'
].
sum
(
axis
=
tuple
(
self
.
attrs
[
'dim'
]))
}
# reduce doesn't support float64 in cinn.
self
.
enable_cinn
=
False
self
.
enable_cinn
=
True
class
Test3DReduce2
(
Test1DReduce
):
...
...
@@ -752,8 +745,7 @@ class Test3DReduce2(Test1DReduce):
self
.
outputs
=
{
'Out'
:
self
.
inputs
[
'X'
].
sum
(
axis
=
tuple
(
self
.
attrs
[
'dim'
]))
}
# reduce doesn't support float64 in cinn.
self
.
enable_cinn
=
False
self
.
enable_cinn
=
True
class
Test3DReduce3
(
Test1DReduce
):
...
...
@@ -766,8 +758,7 @@ class Test3DReduce3(Test1DReduce):
self
.
outputs
=
{
'Out'
:
self
.
inputs
[
'X'
].
sum
(
axis
=
tuple
(
self
.
attrs
[
'dim'
]))
}
# reduce doesn't support float64 in cinn.
self
.
enable_cinn
=
False
self
.
enable_cinn
=
True
class
Test8DReduce0
(
Test1DReduce
):
...
...
@@ -800,8 +791,7 @@ class TestKeepDimReduce(Test1DReduce):
axis
=
tuple
(
self
.
attrs
[
'dim'
]),
keepdims
=
self
.
attrs
[
'keep_dim'
]
)
}
# reduce doesn't support float64 in cinn.
self
.
enable_cinn
=
False
self
.
enable_cinn
=
True
class
TestKeepDim8DReduce
(
Test1DReduce
):
...
...
@@ -897,8 +887,7 @@ class TestReduceSumWithDimOne(OpTest):
axis
=
tuple
(
self
.
attrs
[
'dim'
]),
keepdims
=
True
)
}
# reduce doesn't support float64 in cinn
self
.
enable_cinn
=
False
self
.
enable_cinn
=
True
def
test_check_output
(
self
):
self
.
check_output
()
...
...
@@ -919,8 +908,7 @@ class TestReduceSumWithNumelOne(OpTest):
axis
=
tuple
(
self
.
attrs
[
'dim'
]),
keepdims
=
False
)
}
# reduce doesn't support float64 in cinn
self
.
enable_cinn
=
False
self
.
enable_cinn
=
True
def
test_check_output
(
self
):
self
.
check_output
()
...
...
@@ -937,8 +925,7 @@ class TestReduceAll(OpTest):
self
.
inputs
=
{
'X'
:
np
.
random
.
random
((
100
,
1
,
1
)).
astype
(
"float64"
)}
self
.
attrs
=
{
'reduce_all'
:
True
,
'keep_dim'
:
False
}
self
.
outputs
=
{
'Out'
:
self
.
inputs
[
'X'
].
sum
()}
# reduce doesn't support float64 in cinn
self
.
enable_cinn
=
False
self
.
enable_cinn
=
True
def
test_check_output
(
self
):
self
.
check_output
()
...
...
@@ -955,8 +942,7 @@ class TestReduceAllFp32(OpTest):
self
.
inputs
=
{
'X'
:
np
.
random
.
random
((
100
,
1
,
1
)).
astype
(
"float32"
)}
self
.
attrs
=
{
'reduce_all'
:
True
,
'keep_dim'
:
False
}
self
.
outputs
=
{
'Out'
:
self
.
inputs
[
'X'
].
sum
()}
# reduce doesn't support float64 in cinn
self
.
enable_cinn
=
False
self
.
enable_cinn
=
True
def
test_check_output
(
self
):
self
.
check_output
()
...
...
@@ -973,7 +959,7 @@ class Test1DReduceWithAxes1(OpTest):
self
.
inputs
=
{
'X'
:
np
.
random
.
random
(
100
).
astype
(
"float64"
)}
self
.
attrs
=
{
'dim'
:
[
0
],
'keep_dim'
:
False
}
self
.
outputs
=
{
'Out'
:
self
.
inputs
[
'X'
].
sum
(
axis
=
0
)}
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
def
test_check_output
(
self
):
self
.
check_output
(
check_prim
=
True
)
...
...
@@ -996,6 +982,7 @@ class TestReduceWithDtype(OpTest):
'out_dtype'
:
int
(
convert_np_dtype_to_dtype_
(
np
.
float64
)),
}
)
# cinn op_mapper not support in_dtype/out_dtype attr
self
.
enable_cinn
=
False
def
test_check_output
(
self
):
...
...
@@ -1025,6 +1012,7 @@ class TestReduceWithDtype1(TestReduceWithDtype):
'out_dtype'
:
int
(
convert_np_dtype_to_dtype_
(
np
.
float64
)),
}
)
# cinn op_mapper not support in_dtype/out_dtype attr
self
.
enable_cinn
=
False
def
test_check_output
(
self
):
...
...
@@ -1048,6 +1036,7 @@ class TestReduceWithDtype2(TestReduceWithDtype):
'out_dtype'
:
int
(
convert_np_dtype_to_dtype_
(
np
.
float64
)),
}
)
# cinn op_mapper not support in_dtype/out_dtype attr
self
.
enable_cinn
=
False
def
test_check_output
(
self
):
...
...
python/paddle/fluid/tests/unittests/test_slice_op.py
浏览文件 @
bb5dd203
...
...
@@ -34,7 +34,7 @@ class TestSliceOp(OpTest):
self
.
op_type
=
"slice"
self
.
prim_op_type
=
"prim"
self
.
python_api
=
paddle
.
slice
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
self
.
config
()
self
.
inputs
=
{
'Input'
:
self
.
input
}
self
.
outputs
=
{
'Out'
:
self
.
out
}
...
...
@@ -74,7 +74,7 @@ class TestCase1(TestSliceOp):
class
TestCase2
(
TestSliceOp
):
def
config
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
self
.
input
=
np
.
random
.
random
([
3
,
4
,
5
,
6
]).
astype
(
"float64"
)
self
.
starts
=
[
-
3
,
0
,
2
]
self
.
ends
=
[
3
,
100
,
-
1
]
...
...
@@ -114,7 +114,7 @@ class TestSliceZerosShapeTensor(OpTest):
# 1.2 with attr(decrease)
class
TestSliceOp_decs_dim
(
OpTest
):
def
setUp
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
self
.
op_type
=
"slice"
self
.
prim_op_type
=
"prim"
self
.
python_api
=
paddle
.
slice
...
...
@@ -149,7 +149,7 @@ class TestSliceOp_decs_dim(OpTest):
class
TestSliceOp_decs_dim_2
(
TestSliceOp_decs_dim
):
def
config
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
self
.
input
=
np
.
random
.
random
([
3
,
4
,
5
,
6
]).
astype
(
"float64"
)
self
.
starts
=
[
1
,
0
,
2
]
self
.
ends
=
[
2
,
1
,
4
]
...
...
@@ -161,7 +161,7 @@ class TestSliceOp_decs_dim_2(TestSliceOp_decs_dim):
class
TestSliceOp_decs_dim_3
(
TestSliceOp_decs_dim
):
def
config
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
self
.
input
=
np
.
random
.
random
([
3
,
4
,
5
,
6
]).
astype
(
"float64"
)
self
.
starts
=
[
-
1
,
0
,
2
]
self
.
ends
=
[
1000000
,
1
,
4
]
...
...
@@ -185,7 +185,7 @@ class TestSliceOp_decs_dim_4(TestSliceOp_decs_dim):
class
TestSliceOp_decs_dim_5
(
TestSliceOp_decs_dim
):
def
config
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
self
.
input
=
np
.
random
.
random
([
3
,
4
,
5
,
6
]).
astype
(
"float64"
)
self
.
starts
=
[
-
1
]
self
.
ends
=
[
1000000
]
...
...
@@ -198,7 +198,7 @@ class TestSliceOp_decs_dim_5(TestSliceOp_decs_dim):
# test_6 with test_2 with test_3
class
TestSliceOp_decs_dim_6
(
TestSliceOp_decs_dim
):
def
config
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
self
.
input
=
np
.
random
.
random
([
3
,
4
,
5
,
6
]).
astype
(
"float64"
)
self
.
starts
=
[
0
,
1
,
2
,
3
]
self
.
ends
=
[
1
,
2
,
3
,
4
]
...
...
@@ -484,7 +484,7 @@ class TestSliceOp_starts_OneTensor_ends_ListTensor(OpTest):
)
class
TestFP16
(
OpTest
):
def
setUp
(
self
):
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
self
.
op_type
=
"slice"
self
.
prim_op_type
=
"prim"
self
.
python_api
=
paddle
.
slice
...
...
python/paddle/fluid/tests/unittests/test_softmax_op.py
浏览文件 @
bb5dd203
...
...
@@ -73,7 +73,7 @@ class TestSoftmaxOp(OpTest):
'use_cudnn'
:
self
.
use_cudnn
,
'use_mkldnn'
:
self
.
use_mkldnn
,
}
self
.
enable_cinn
=
Fals
e
self
.
enable_cinn
=
Tru
e
def
init_kernel_type
(
self
):
pass
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录