Just caught up to this 😅 : so OpenAI's Triton is a pythonesque DSL for exposing hardware primitives using a "tile" paradigm.
Handles hardware for you
- - numpy : CPU only
- - tensorflow, pytorch : anyPU
You handle hardware :
- - CUDA : nVideaPU only
- - triton / C/++ / SYCL / openCL : anyPU
Triton is a dialect of MLIR@LLVM ...
No comments :
Post a Comment