Untitled
unknown
plain_text
a year ago
507 B
4
Indexable
@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"]): # extra annotations for the function T.func_attr({"global_symbol": "main", "tir.noalias": True}) for i in range(128): with T.block("C"): # declare a data parallel iterator on spatial domain vi = T.axis.spatial(128, i) C[vi] = A[vi] + B[vi]
Editor is loading...
Leave a Comment