Any thoughts on the best way to do this?
I gave it a shot here: https://gist.github.com/bwasti/a9d3b82d06f0de7df34a04e0c1ae4764
but didn’t make much headway – I hit a hard to decipher error in the Bind logic:
RuntimeError: [23:10:17] /home/bwasti/local/pytorch_tvm/tvm/src/runtime/module_util.cc:54: Check failed: ret == 0 (-1 vs. 0) : Assert fail: (int32(arg0.shape[0]) ==
33554432), Argument arg0.shape[0] has an unsatisfied constraint