Thanks for pointing out to this meta device. I was not aware of it. Very interesting indeed!
Though I guess for this specific question it’s maybe not super useful as the cost of running a kernel on a singleton tensor is fine. And I just need to run this once in a while, so not terrible even if it takes half an hour to go through all the ops.
I was really hoping the typing rules were available somewhere… Since there are all these yaml files with a lot of info, I was hopeful