Custom datatypes in AutoScheduler?

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