提交 c1a53d7d authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

SparseMultiply: Cleanup Ops

* Handle static shape * Rename to more readable Op classes * Simplify perform
上级 2e4e3095
差异被折叠。
...@@ -75,7 +75,8 @@ def test_local_mul_s_d(): ...@@ -75,7 +75,8 @@ def test_local_mul_s_d():
f = pytensor.function(inputs, smath.mul_s_d(*inputs), mode="CVM") f = pytensor.function(inputs, smath.mul_s_d(*inputs), mode="CVM")
assert not any( assert not any(
isinstance(node.op, smath.MulSD) for node in f.maker.fgraph.toposort() isinstance(node.op, smath.SparseDenseMultiply)
for node in f.maker.fgraph.toposort()
) )
...@@ -92,7 +93,8 @@ def test_local_mul_s_v(): ...@@ -92,7 +93,8 @@ def test_local_mul_s_v():
f = pytensor.function(inputs, smath.mul_s_v(*inputs), mode="CVM") f = pytensor.function(inputs, smath.mul_s_v(*inputs), mode="CVM")
assert not any( assert not any(
isinstance(node.op, smath.MulSV) for node in f.maker.fgraph.toposort() isinstance(node.op, smath.SparseDenseVectorMultiply)
for node in f.maker.fgraph.toposort()
) )
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论