[Mlir-commits] [mlir] [mlir][gpu] Support Cluster of Thread Blocks in `gpu.launch_func` (PR #72871)
Mehdi Amini
llvmlistbot at llvm.org
Mon Nov 20 20:53:41 PST 2023
================
@@ -121,6 +121,26 @@ class ConvertOpToGpuRuntimeCallPattern : public ConvertOpToLLVMPattern<OpTy> {
llvmPointerType, /* void **extra */
llvmInt64Type /* size_t paramsCount */
}};
+ FunctionCallBuilder launchClusterKernelCallBuilder = {
+ "mgpuLaunchClusterKernel",
+ llvmVoidType,
+ {
+ llvmPointerType, /* void* f */
+ llvmIntPtrType, /* intptr_t clusterXDim */
+ llvmIntPtrType, /* intptr_t clusteryDim */
+ llvmIntPtrType, /* intptr_t clusterZDim */
+ llvmIntPtrType, /* intptr_t gridXDim */
+ llvmIntPtrType, /* intptr_t gridyDim */
+ llvmIntPtrType, /* intptr_t gridZDim */
+ llvmIntPtrType, /* intptr_t blockXDim */
+ llvmIntPtrType, /* intptr_t blockYDim */
+ llvmIntPtrType, /* intptr_t blockZDim */
+ llvmInt32Type, /* unsigned int sharedMemBytes */
+ llvmPointerType, /* void *hstream */
+ llvmPointerType, /* void **kernelParams */
+ llvmPointerType, /* void **extra */
+ llvmInt64Type /* size_t paramsCount */
+ }};
----------------
joker-eph wrote:
We don't have to support the deprecated path with new features right now I believe.
https://github.com/llvm/llvm-project/pull/72871
More information about the Mlir-commits
mailing list