[GitHub] [tvm] tkonolige commented on issue #10831: [Bug][TIR][Script] TVM complains about undefined variables when building PrimFunc

[GitBox <gi...@apache.org>]

tkonolige commented on issue #10831:
URL: https://github.com/apache/tvm/issues/10831#issuecomment-1086287433


   ```
   @T.prim_func
   def peak_bandwidth(a: T.handle, b: T.handle) -> None:
       N = T.var("int32")
       A = T.match_buffer(a, [N], "float32")
       B = T.match_buffer(b, [8 * 4], "float32")
       C = T.alloc_buffer([8 * 4], "float32")
       for r in range(1234):
           for i in range(N // (8 * 4)):
               for l in T.unroll(4):
                   for j in T.vectorized(8):
                       C[l * 8 + j] = A[i * 8 * 4 + l * 4 + j]
                       B[l * 8 + j] = B[l * 8 + j] + C[(3 - l) * 8 + j]
   ```
   
   I get a similar error here.


-- 
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-unsubscribe@tvm.apache.org

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