The modified code looks good to me. Maybe you can double check what's going on inside those registrations, e.g. checking whether the mapping in c++ side is indeed updated when `register` is called
--- [Visit Topic](https://discuss.tvm.apache.org/t/custom-datatypes-in-autoscheduler/12377/6) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/77c7a34dd4e512d3fcf75038409a162297ac902bd2ae2d45cc67077626214b8e).