以下の内容はhttps://vengineer.hatenablog.com/entry/71475425より取得しました。


Tensor Comprehensions



FacebookTensor Comprehensions をアナウンスしました。


この Tensor Comprehensions では、下図(引用します)に、HalideIRを採用しているようです。

https://research.fb.com/wp-content/uploads/2018/02/tc_scheme.png

コード生成部は独自です。

CPUへはLLVM経由でコード生成していますが、
GPU(CUDA)へはLLVM(PTX)ではなく、CUDAのソースコードを生成し、
NVRTCライブラリを使ってソースコードコンパイルしています。

また、ATen: A TENsor libraryも使うことができます。

TVMでもHalideIRを取り込んでいます。

Halideは画像処理をターゲットにしていますが、ディープラーニングにも利用が進んできましたね。

追記)、2018.03.01
For our initial system,
we focus on the generation of CUDA code
because NVIDIA GPUs dominate the hardware landscape for training deep neural networks.
とある。



以上の内容はhttps://vengineer.hatenablog.com/entry/71475425より取得しました。
このページはhttp://font.textar.tv/のウェブフォントを使用してます

不具合報告/要望等はこちらへお願いします。
モバイルやる夫Viewer Ver0.14