Re:Prefix sum on portable compute shaders
by masahi on Wednesday December 01, 2021 @07:38AM
(#29403108)
TVM, similar to IREE, also has a good support for Vulkan. It compiles a tensor DSL written in Python into SPIR-V. AMD is using it in production for running deep learning models on their APU.
https://github.com/apache/tvm