polygeistのコードベースを確認する

[yukarinoki]

確認するコミット

https://github.com/llvm/Polygeist/tree/cgo24

Retargetingのcommitによって確認する。

[yukarinoki]

polygeist側にはほぼ編集がない
llvm/Polygeist@main...cgo24

llvm側に編集があるのだろうか？？

[yukarinoki]

llvm側の編集はこれだと思う

https://github.com/ivanradanov/llvm-project/tree/0b9310c6e4416ee48c07edfef81144e22850dfe7

[yukarinoki]

うーむ、これだという編集が見えないな。
gpu-kernel-emit-coarsened-alternativesで調べる。

[yukarinoki]

以下のpull requestが、cgo24の主要なコミットである。

• Implement thread coarsening for GPU kernels
  Implement thread coarsening for GPU kernels llvm/Polygeist#352

以下のpull requestも気になる。pluto-optでしょうか。

• Integrate polymer's pluto-opt into polygeist pipeline
  Integrate polymer's pluto-opt into polygeist pipeline llvm/Polygeist#409

[yukarinoki]

以下の論文をもう一回見てみよう！

[yukarinoki]

Polygeistのパイプラインは、図1のように4つのコンポーネントで構成されています：

1. フロントエンド：CまたはC++コードからSCFループレベルでMLIRに入ることを可能にします（セクションIII-A）。
2. 前処理ステップ：MLIR内でアフィンダイアレクトに引き上げます（セクションIII-B）。
3. 多面体スケジューラ：OpenSCoPへの往復を介してプログラムのアフィン部分を処理し、新しい文分割ヒューリスティックによって制御されるPluto変換を実行します（セクションIII-CおよびIII-D）。
4. バックエンド：ポストプロセッシングのMLIR最適化を実行し、最終的に実行可能ファイルにロワリングします（セクションIII-E）。

[yukarinoki]

以下の論文が大事である

https://inria.hal.science/hal-04711965/

[yukarinoki]

やるべきこと

1. affine MLIRが出力されることを確認する
   行列積がいいかな

2. GPUのIRを出力するための方法を探る。

[yukarinoki]

include/polygeist/Passes/Passes.h

変換パスが書いてある

std::unique_ptr<Pass>
createConvertParallelToGPUPass1(std::string arch = "sm_60");
std::unique_ptr<Pass>
createConvertParallelToGPUPass2(bool emitGPUKernelLaunchBounds = true);

include/polygest/Passes/Passes.td

TableGenで書かれた、Passの定義集
Parallel To GPUのpassはivanが書いてくれている

だから、僕はaffine to parallelを書いたら良いのだろうか？？？
まあ

lib/polygeist/Ops.cpp

ここにOpの変換の定義と、Opのクラスのコンストラクタ（ファクトリ？）が書いてある。

ivanが書いてあるのは、AlternativeOpに対する、FlattenAlternativesという変換がある。多分、Threadとかblockとかの調節をやっているのかもしれない

lib/polygeist/Passes/CMakeLists.txt

ここにpass関連のビルド定義が書いてある。

if(POLYGEIST_ENABLE_CUDA)
  find_package(CUDA)
  enable_language(CUDA)

  # Enable gpu-to-cubin pass.
  target_compile_definitions(obj.MLIRPolygeistTransforms
    PRIVATE
    POLYGEIST_ENABLE_CUDA=1
  )

  # Add CUDA headers includes and the libcuda.so library.
  target_include_directories(obj.MLIRPolygeistTransforms
    PRIVATE
    ${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES}
  )

  find_library(CUDA_DRIVER_LIBRARY cuda)

  target_link_libraries(MLIRPolygeistTransforms
    PRIVATE
    ${CUDA_DRIVER_LIBRARY}
  )

endif()

tools/cgeist/driver.cc

ここら辺に動かし方が書いてある
これが重要な気がするなぁ