Fix test case in test_tir_transform_split_host_device

slyubomirsky · slyubomirsky · commit 645d356b0a7f · 2023-07-18T20:50:14.000-04:00
diff --git a/tests/python/unittest/test_tir_transform_split_host_device.py b/tests/python/unittest/test_tir_transform_split_host_device.py
@@ -152,7 +152,7 @@ def main(n: T.int32):
                 err = mod.main_kernel(n)
                 assert err == 0, "Error executing compute kernel"
 
-            @T.prim_func
+            @T.prim_func(private=True)
             def main_kernel(n: T.int32) -> T.int32:
                 T.func_attr(
                     {

Original file line number	Diff line number	Diff line change
`@@ -152,7 +152,7 @@ def main(n: T.int32):`
`152`	`152`	`err = mod.main_kernel(n)`
`153`	`153`	`assert err == 0, "Error executing compute kernel"`
`154`	`154`
`155`		`- @T.prim_func`
	`155`	`+ @T.prim_func(private=True)`
`156`	`156`	`def main_kernel(n: T.int32) -> T.int32:`
`157`	`157`	`T.func_attr(`
`158`	`158`	`{`