Revert "CONST(VIEW(DEVICE)) (#8365)" (#8372)

This reverts commit 83284985f0.
This commit is contained in:
qazal
2024-12-22 04:44:34 +02:00
committed by GitHub
parent 83284985f0
commit 514a6740e4
4 changed files with 22 additions and 17 deletions

View File

@@ -1987,7 +1987,7 @@ class TestBigGraph(unittest.TestCase):
check_schedule(x, 1)
tensor_const_pm = PatternMatcher([
(UPat(Ops.CONST, src=(UPat(Ops.VIEW, src=(UPat(Ops.DEVICE),)))), lambda: True),
(UPat(Ops.VIEW, src=(UPat(Ops.DEVICE), UPat(Ops.CONST, src=()))), lambda: True),
(UPat(Ops.VIEW, src=(UPat(Ops.DEVICE), UPat(Ops.BIND, src=(UPat(Ops.DEFINE_VAR), UPat(Ops.CONST))))), lambda: True),
])
class TestConst(unittest.TestCase):