From 21d32723131ffaae024335269106aa7c86ddb53f Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Thu, 7 Mar 2024 14:10:47 +0100
Subject: [PATCH] fix constant elim tests
---
.../transformations/test_constant_elimination.py | 10 +++++++---
1 file changed, 7 insertions(+), 3 deletions(-)
diff --git a/tests/nbackend/transformations/test_constant_elimination.py b/tests/nbackend/transformations/test_constant_elimination.py
index da89ab044..b2ac6fc5a 100644
--- a/tests/nbackend/transformations/test_constant_elimination.py
+++ b/tests/nbackend/transformations/test_constant_elimination.py
@@ -1,3 +1,4 @@
+from pystencils.backend.kernelcreation import KernelCreationContext
from pystencils.backend.ast.expressions import PsExpression, PsConstantExpr
from pystencils.backend.symbols import PsSymbol
from pystencils.backend.constants import PsConstant
@@ -21,7 +22,8 @@ i12 = PsExpression.make(PsConstant(12, Int(32)))
def test_idempotence():
- elim = EliminateConstants()
+ ctx = KernelCreationContext()
+ elim = EliminateConstants(ctx)
expr = f42 * (f1 + f0) - f0
result = elim(expr)
@@ -42,7 +44,8 @@ def test_idempotence():
def test_int_folding():
- elim = EliminateConstants()
+ ctx = KernelCreationContext()
+ elim = EliminateConstants(ctx)
expr = (i1 * x + i1 * i3) + i1 * i12
result = elim(expr)
@@ -54,7 +57,8 @@ def test_int_folding():
def test_zero_dominance():
- elim = EliminateConstants()
+ ctx = KernelCreationContext()
+ elim = EliminateConstants(ctx)
expr = (f0 * x) + (y * f0) + f1
result = elim(expr)
--
GitLab