-
在 2.4.3.5. 为什么块需要额外附加的信息 中提到:
例子是当迭代器为128的轴被绑定在一个range(127)的i循环时会产生报错,在实际使用场景下发现,如果后续有计算代码则不会报错,后续没有计算代码会报错,e.g. import tvm
from tvm.script import tir as T
@tvm.script.ir_module
class MyModule:
@T.prim_func
def main(A: T.Buffer[128, "float32"],
B: T.Buffer[128, "float32"],
C: T.Buffer[128, "float32"]):
for i in range(127): # 127 for loop
with T.block("C"):
vi = T.axis.spatial(128, i) # Bind to 128 iterator
C[vi] = A[vi] + B[vi] # 不加会报错,加上不会报错 不是很理解后续跟有计算时报错为什么被屏蔽了呢?实际可以被执行起来吗?(因为本人LLVM后端出现问题无法生成可调用函数所以无法继续向下尝试) |
Beta Was this translation helpful? Give feedback.
Answered by
Hzfengsy
Jul 20, 2022
Replies: 1 comment
-
相关讨论:#24 |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
CookieLau
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
相关讨论:#24