Tiramisuのbenchmark (DNN)

@Vengineerの戯言 : Twitter
SystemVerilogの世界へようこそ、すべては、SystemC v0.9公開から始まった

Halide勉強会で発表した Tiramisu の benchmark (DNN) がいろいろサポートされていました。

vengineer.hatenablog.com

layers の下には、下記の5つのフォルダーがありますね。

• bn/cpu
• convolution
• maxpool/cpu
• relu/cpu
• transpose_conv

MKL と MKLDNN と比較していますね。

blocks の下には、

• Conv-ReLU-MaxPool/cpu 
• Conv-Relu-FC-Softmax/cpu/dense 
• DenseNetBlock/cpu/dense 
• LSTM
• Resize-Conv-ReLU-MaxPool/cpu
• conv2/cpu/dense
• fusedresNet/cpu/dense
• fusedresNet_inference/cpu
• vggBlock/cpu

cpu で終わっているフォルダーの下に、dense と sparse というフォルダーがあります。

LSTM に関しては、以下の5つのフォルダーがありますね。

• cpu
• cpu_lib
• cpu_lib_sparse
• gpu
• gpu_lib

この LSTM だけ、GPU 版がありますね。