slyubomirsky commented on code in PR #16569:
URL: https://github.com/apache/tvm/pull/16569#discussion_r1506773164


##########
tests/python/codegen/test_inject_ptx_ldg32.py:
##########
@@ -20,7 +20,8 @@
 import tvm.testing
 
 
-@T.prim_func
+# A_local is undefined
+@T.prim_func(check_well_formed=False)

Review Comment:
   Hm, possible parsing bug? I get this error:
   ```python
   ValueError: Block frame or PrimFunc frame not find. Please ensure 
'T.alloc_buffer' is called under T.block() or T.prim_func()
    --> 
/home/slyubomirsky/code/tvm/tests/python/codegen/test_inject_ptx_ldg32.py:31:15
       |  
    31 |      A_local = T.alloc_buffer((32), "float32", scope="local")
       |                ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
   ```
   It is being done inside a PrimFunc, so I don't know why we would get that 
error. I'll add a block and see if the example will work regardless.



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscr...@tvm.apache.org

For queries about this service, please contact Infrastructure at:
us...@infra.apache.org

Reply via email to