提交 eb6fc66e authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Ricardo Vieira

Compare single elements directly in compare_numba_and_py

上级 e4ca40f9
...@@ -127,8 +127,6 @@ def set_test_value(x, v): ...@@ -127,8 +127,6 @@ def set_test_value(x, v):
def compare_shape_dtype(x, y): def compare_shape_dtype(x, y):
(x,) = x
(y,) = y
return x.shape == y.shape and x.dtype == y.dtype return x.shape == y.shape and x.dtype == y.dtype
...@@ -286,7 +284,7 @@ def compare_numba_and_py( ...@@ -286,7 +284,7 @@ def compare_numba_and_py(
for j, p in zip(numba_res, py_res): for j, p in zip(numba_res, py_res):
assert_fn(j, p) assert_fn(j, p)
else: else:
assert_fn(numba_res, py_res) assert_fn(numba_res[0], py_res[0])
return pytensor_numba_fn, numba_res return pytensor_numba_fn, numba_res
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论