Skip to content

Commit 858c4c0

Browse files
committed
Add another test
1 parent 2770429 commit 858c4c0

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

test/squin/test_mult_rewrite.py

+7
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,12 @@ def nested_rmul():
139139
y = squin.op.y()
140140
return 2 * x * y
141141

142+
@squin.kernel
143+
def nested_rmul2():
144+
x = squin.op.x()
145+
y = squin.op.y()
146+
return 2 * (x * y)
147+
142148
@squin.kernel
143149
def nested_lmul():
144150
x = squin.op.x()
@@ -156,4 +162,5 @@ def check_stmt_type(code, typ):
156162
check_stmt_type(simple_lmul.code, complex)
157163
check_stmt_type(simple_rmul.code, float)
158164
check_stmt_type(nested_rmul.code, int)
165+
check_stmt_type(nested_rmul2.code, int)
159166
check_stmt_type(nested_lmul.code, complex)

0 commit comments

Comments
 (0)