提交 566af64d authored 作者: ricardoV94's avatar ricardoV94 提交者: Ricardo Vieira

Use bitset to check ancestors more efficiently

上级 dc1e3b9c
......@@ -301,7 +301,8 @@ def test_debugprint():
Gemv_op_name = "CGemv" if pytensor.config.blas__ldflags else "Gemv"
exp_res = dedent(
r"""
Composite{(i2 + (i0 - i1))} 4
Composite{(i0 + (i1 - i2))} 4
├─ A
├─ ExpandDims{axis=0} v={0: [0]} 3
"""
f" │ └─ {Gemv_op_name}{{inplace}} d={{0: [0]}} 2"
......@@ -313,17 +314,16 @@ def test_debugprint():
│ ├─ B
│ ├─ <Vector(float64, shape=(?,))>
│ └─ 0.0
├─ D
└─ A
└─ D
Inner graphs:
Composite{(i2 + (i0 - i1))}
Composite{(i0 + (i1 - i2))}
← add 'o0'
├─ i2
└─ sub
├─ i0
└─ i1
└─ sub
├─ i1
└─ i2
"""
).lstrip()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论