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