2 ms·

Probably should make this into a backend of torch.compile

• zhihaojia 1y ago
  Yes, it would be a lot of fun if MPK can enable torch.compile to generate megakernels. Torch-generated kernels are currently too slow for latency-sensitive workloads.