提交 b99dea2f authored 作者: Brandon T. Willard's avatar Brandon T. Willard

Add xfail for uncovered local_subtensor_make_vector index type condition

上级 1c730796
......@@ -2781,6 +2781,22 @@ class TestLocalSubtensorMakeVector:
r = f(0, 1, 2)
assert r[0] == 0 and r[1] == 2
@pytest.mark.xfail(
reason="local_subtensor_make_vector doesn't handle all index cases"
)
def test_MakeVector_idx(self):
x, y, z, q = tensor.lscalars("xyzq")
v = make_vector(x, y, z)
q = make_vector(0, 2)
f = function([x, y, z], v[q], mode=mode_opt)
prog = f.maker.fgraph.toposort()
assert len(prog) == 1
assert isinstance(prog[0].op, MakeVector)
assert len(prog[0].inputs) == 2
r = f(0, 1, 2)
assert r[0] == 0 and r[1] == 2
def test_stack_trace(self):
x, y, z = tensor.lscalars("xyz")
v = make_vector(x, y, z)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论