HomePhabricator

[mlir][gpu] Refactor ConvertGpuLaunchFuncToCudaCalls pass.

Authored by whchung on May 18 2020, 11:01 AM.

Description

[mlir][gpu] Refactor ConvertGpuLaunchFuncToCudaCalls pass.

Due to similar APIs between CUDA and ROCm (HIP),
ConvertGpuLaunchFuncToCudaCalls pass could be used on both platforms with some
refactoring.

In this commit:

  • Migrate ConvertLaunchFuncToCudaCalls from GPUToCUDA to GPUCommon, and rename.
  • Rename runtime wrapper APIs be platform-neutral.
  • Let GPU binary annotation attribute be specifiable as a PassOption.
  • Naming changes within the implementation and tests.

Subsequent patches would introduce ROCm-specific tests and runtime wrapper
APIs.

Differential Revision: https://reviews.llvm.org/D80167