Hi! As I am following the d2l-tvm documentation, I would like to know if there is a way to get the schedule from a tvm.build. For instance, the simple tvm_vector_add has
def tvm_vector_add(n, dtype):
A = te.placeholder((n,), dtype=dtype)
B = te.placeholder((n,), dtype=dtype)
C = te.compute(A.shape, lambda i: A[i] + B[i])
print('expression dtype:', A.dtype, B.dtype, C.dtype)
s = te.create_schedule(C.op)
return tvm.build(s, [A, B, C])
If I were to do
mod = tvm_vector_add(100, "float32")
Is there a way to get the schedule (s in tvm_vector_add) from mod?
Thank you.
tvm 0.13.dev0
python 3.8