huy*_*u11 3 opencl metalkit coreml
我打算将我的项目的某些GPU内核从OpenCL / Cuda转换为Metal,以便在Apple设备上运行我的应用程序。目前,我的项目是完全用C / C ++编写的。经过研究后,我认为我需要弄清楚Swift或Objective-C。但老实说,我不确定这些东西,因为用于计算和深度学习的金属语言是相当新的。
我知道有一个名为“ CoreML ” 的库,但是我的应用程序需要一些自定义内核。我的问题:在我的情况下,处理Apple设备低级API的最佳方法是什么?
金属着色语言是C ++的版本。在将OpenCL或CUDA内核移植到Metal方面,我没有太多麻烦。
Core ML仅支持有限的一组图层。您可以编写自己的自定义图层,其中包括编写CPU版本和可选的GPU版本(金属语言)。
我写了一篇关于此的博客文章:http : //machinethink.net/blog/coreml-custom-layers/