From 405a3d4799ef3cae66b13dea46fc93d32f4c03df Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Tue, 15 Nov 2022 16:37:18 +0900 Subject: [PATCH] fix te create_primfunc test --- tests/python/unittest/test_te_create_primfunc.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/python/unittest/test_te_create_primfunc.py b/tests/python/unittest/test_te_create_primfunc.py index b59880758e5d..7b8173d0b2d9 100644 --- a/tests/python/unittest/test_te_create_primfunc.py +++ b/tests/python/unittest/test_te_create_primfunc.py @@ -390,6 +390,7 @@ def expected_layout_attr( C[x, y] = C[x, y] + A[x, k] * B[y, k] for i0, i1 in T.grid(128, 128): with T.block("D"): + T.block_attr({"layout_free_placeholders": [C]}) x, y = T.axis.remap("SS", [i0, i1]) D[x, y] = C[x, y] + T.float32(1)