I have recently started looking at TVM and VTA and would like to see the instructions that are generated for each convolutional layer.
I started by running
test_benchmark_topi_conv2d.py example and later modified the code to print debugging information according to this answer (the rest of the code is not touched):
# I added the following line with vta.build_config(debug_flag=2): if "vta" in target.keys: mod = vta.build(s, [data, kernel, bias, res], target=target, target_host=env.target_host, name="conv2d") else: mod = tvm.build(s, [data, kernel, bias, res], target=target, target_host=env.target_host, name="conv2d")
However, running the new code returns the following error:
module 'tvm.make' has no attribute 'stmt_seq'
I would really appreciate it if you let me know how to print debugging information.