Open deep learning compiler stack for cpu, gpu and specialized accelerators
javascript
machine-learning
performance
deep-learning
metal
compiler
gpu
vulkan
opencl
tensor
spirv
rocm
tvm
-
Updated
Nov 9, 2020 - Python
A few of the reference kernels in iree/hal/vmla/op_kernels_generic.h have particularly poor performance. While we expect the LLVM ahead-of-time backend to be more viable as a deployment target, having a faster reference backend is still generally useful.
Some of the slow kernels are already labeled:
https://github