PyTorch + TVM

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

PyTorch + XLA に続き、TVMとも繋いでいる。。。と。

2018年10月9日、PyTorchがTPUで動く

2018年11月16日、PyTorch + XLA のソースコードが公開された

2018年12月1日、「PyTorch + XLA」のソースコード解析 (その1)
2018年12月24日、「PyTorch + XLA」のソースコード解析 (その2)
2018年12月25日、「PyTorch + XLA」のソースコード解析 (その3)

によると、PyTorch IR から Lowerにて、TVMに。

PyBind11にて、C++コードをPythonと繋いでいます (torch_tvm/register.cpp)。

TVMCompilerのrunメソッドが呼ばれるみたい。

PyTorchのグラフを TVM/Relayのグラフに変換して、

　・build_f(func, "llvm", "llvm -mcpu=core-avx2");

で、TVM => LLVM にて、AVX2にて最適化。。。。あー、決まったコードが出来るか。。。

P.S
なんか、自分のブログに教えられた、2018年8月15日、PyTorch/MXNet と TVM と繋ぐには？