Compiling NumPy code into C++ or CUDA via torch.compile
pytorch.org
Quansight engineers have implemented support for tracing through NumPy code via torch.compile in PyTorch 2.1. This feature leverages PyTorch’s compiler to generate efficient fused vectorized code without having to modify your original NumPy code. Even more, it also allows for executing NumPy code on CUDA just by running it through torch.compile under torch.device(
  • kernelPanic@lemmy.ml
    ·
    1 year ago

    I tried torch compile before, it is nice but resolving cpp errors in Python is a pain. I am more excited about mojo