Comment by qihqi

Probably should make this into a backend of torch.compile