第3课中T.grid 和 T.axis.spatial对于循环范围的指定是否重复了？

[bigpieit]

for i, j, k in T.grid(128, 128, 128): 已经指定了循环范围，为何T.axis.spatial(128, i) 又要再次指认循环范围呢？仅指认为spatial dimension可以吗？

@tvm.script.ir_module
class MyModule:
    @T.prim_func
    def mm_relu(A: T.Buffer[(128, 128), "float32"], 
                B: T.Buffer[(128, 128), "float32"], 
                C: T.Buffer[(128, 128), "float32"]):
        T.func_attr({"global_symbol": "mm_relu", "tir.noalias": True})
        Y = T.alloc_buffer((128, 128), dtype="float32")
        for i, j, k in T.grid(128, 128, 128):
            with T.block("Y"):
                vi = T.axis.spatial(128, i)
                vj = T.axis.spatial(128, j)
                vk = T.axis.reduce(128, k)
                with T.init():
                    Y[vi, vj] = T.float32(0)
                Y[vi, vj] = Y[vi, vj] + A[vi, vk] * B[vk, vj]
        for i, j in T.grid(128, 128):
            with T.block("C"):
                vi = T.axis.spatial(128, i)
                vj = T.axis.spatial(128, j)             
                C[vi, vj] = T.max(Y[vi, vj], T.float32(0))

[hbsun2113]

再次指定范围就多一次检查，避免中间操作有误，使得边界错误