提交 2e297865 authored 作者: Frederic's avatar Frederic

Fix test in FAST_COMPILE as now we register more opt in fast_compile.

上级 adac885a
......@@ -6430,7 +6430,8 @@ class TestInferShape(utt.InferShapeTester):
self._compile_and_check([advec, bdvec],
[Dot()(advec, bdvec)],
[advec_val, bdvec_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
(Dot, tensor.blas.Dot22,
tensor.blas.Gemv, tensor.blas_c.CGemv))
#mat/mat
admat = dmatrix()
......@@ -6447,14 +6448,16 @@ class TestInferShape(utt.InferShapeTester):
self._compile_and_check([advec, bdmat],
[Dot()(advec, bdmat)],
[advec_val, bdmat_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
(Dot, tensor.blas.Dot22,
tensor.blas.Gemv, tensor.blas_c.CGemv))
#mat/vec
admat_val = rand(5, 4)
self._compile_and_check([admat, bdvec],
[Dot()(admat, bdvec)],
[admat_val, bdvec_val],
(Dot, tensor.blas.Gemv, tensor.blas_c.CGemv))
(Dot, tensor.blas.Dot22,
tensor.blas.Gemv, tensor.blas_c.CGemv))
# Split
aivec = ivector()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论