[mlir] create gpu memset op
authorLoren Maggiore <loreno@google.com>
Sat, 4 Sep 2021 06:03:33 +0000 (08:03 +0200)
committerChristian Sigg <csigg@google.com>
Sat, 4 Sep 2021 06:13:04 +0000 (08:13 +0200)
commit361458b1ce890bd3f3575cca82b9bbb01d270e6f
treeba88bd9603cf4ca896dcbb1adaab78f6127e4bb1
parentbb51f76fb179a3d13fb379728070756cc9bf8a12
[mlir] create gpu memset op

Create a gpu memset op and corresponding CUDA and ROCm wrappers.

Reviewed By: herhut, lorenrose1013

Differential Revision: https://reviews.llvm.org/D107548
mlir/include/mlir/Dialect/GPU/GPUOps.td
mlir/lib/Conversion/GPUCommon/GPUToLLVMConversion.cpp
mlir/lib/Dialect/GPU/IR/GPUDialect.cpp
mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
mlir/lib/ExecutionEngine/RocmRuntimeWrappers.cpp
mlir/test/Conversion/GPUCommon/lower-memset-to-gpu-runtime-calls.mlir [new file with mode: 0644]
mlir/test/Dialect/GPU/canonicalize.mlir
mlir/test/Dialect/GPU/invalid.mlir
mlir/test/Dialect/GPU/ops.mlir