Searched refs:kernelFunc (Results 1 – 8 of 8) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
| D | KernelOutlining.cpp | 215 gpu::GPUFuncOp kernelFunc, in convertToLaunchFuncOp() argument
219 launchOp.getLoc(), kernelFunc, launchOp.getGridSizeOperandValues(), in convertToLaunchFuncOp()
278 gpu::GPUModuleOp createKernelModule(gpu::GPUFuncOp kernelFunc, in createKernelModule() argument
286 OperationState state(kernelFunc.getLoc(), in createKernelModule()
288 gpu::GPUModuleOp::build(builder, state, kernelFunc.getName()); in createKernelModule()
291 symbolTable.insert(kernelFunc); in createKernelModule()
293 SmallVector<Operation *, 8> symbolDefWorklist = {kernelFunc}; in createKernelModule()
|
| /external/llvm-project/clang/test/CodeGenSYCL/ |
| D | convergent.cpp | 13 __attribute__((sycl_kernel)) void kernel_single_task(const Func &kernelFunc) { in kernel_single_task() argument
14 kernelFunc(); in kernel_single_task()
|
| /external/llvm-project/clang/test/SemaSYCL/ |
| D | prohibit-thread-local.cpp | 42 kernel_single_task(Func kernelFunc) { kernelFunc(); } in kernel_single_task() argument
|
| D | float128.cpp | 58 __attribute__((sycl_kernel)) void kernel(Func kernelFunc) { in kernel() argument
60 kernelFunc(); in kernel()
|
| D | int128.cpp | 66 __attribute__((sycl_kernel)) void kernel(Func kernelFunc) { in kernel() argument
68 kernelFunc(); in kernel()
|
| /external/llvm-project/mlir/lib/Conversion/SPIRVToLLVM/ |
| D | ConvertLaunchFuncToLLVMCalls.cpp | 180 auto kernelFunc = module.lookupSymbol<LLVM::LLVMFuncOp>(newKernelFuncName); in matchAndRewrite() local
181 if (!kernelFunc) { in matchAndRewrite()
184 kernelFunc = rewriter.create<LLVM::LLVMFuncOp>( in matchAndRewrite()
256 rewriter.replaceOpWithNewOp<LLVM::CallOp>(op, kernelFunc, in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/GPU/IR/ |
| D | GPUDialect.cpp | 101 Operation *kernelFunc = module.lookupSymbol(launchOp.kernel()); in verifyOperationAttribute() local
102 auto kernelGPUFunction = dyn_cast_or_null<gpu::GPUFuncOp>(kernelFunc); in verifyOperationAttribute()
103 auto kernelLLVMFunction = dyn_cast_or_null<LLVM::LLVMFuncOp>(kernelFunc); in verifyOperationAttribute()
107 if (!kernelFunc->getAttrOfType<mlir::UnitAttr>( in verifyOperationAttribute()
431 GPUFuncOp kernelFunc, KernelDim3 gridSize, in build() argument
437 auto kernelModule = kernelFunc->getParentOfType<GPUModuleOp>(); in build()
439 kernelModule.getName(), {builder.getSymbolRefAttr(kernelFunc.getName())}); in build()
|
| /external/llvm-project/mlir/include/mlir/Dialect/GPU/ |
| D | GPUOps.td | 380 OpBuilderDAG<(ins "GPUFuncOp":$kernelFunc, "KernelDim3":$gridSize,
|