Lines Matching +defs:arg +defs:gen

179     for (BlockArgument arg :
181       for (Operation *user : arg.getUsers())
535         for (auto [arg, var] : llvm::zip_equal(blockArgs, operands))
536           moduleTranslation.mapValue(arg, moduleTranslation.lookupValue(var));
828   OwningReductionGen gen =
846   return gen;
1047 /// where the user can continue the code-gen directly afterwards.
1465     // map copyRegion rhs arg
1470     // map copyRegion lhs arg
3476   // kernel arg structure. It primarily becomes relevant in cases like
3633         for (auto [arg, useDevVar] :
3652               moduleTranslation.mapValue(arg, devPtrInfoMap);
3869 // \param arg - This is the generated kernel function argument that
3888 createDeviceArgumentAccessor(MapInfoData &mapData, llvm::Argument &arg,
3914   llvm::Value *v = builder.CreateAlloca(arg.getType(), allocaAS);
3916   if (allocaAS != defaultAS && arg.getType()->isPointerTy())
3919   builder.CreateStore(&arg, v);
4265     for (auto [arg, mapOp] : llvm::zip_equal(mapBlockArgs, mapVars)) {
4269       moduleTranslation.mapValue(arg, mapOpValue);
4339   auto argAccessorCB = [&](llvm::Argument &arg, llvm::Value *input,
4349       retVal = cast<llvm::Value>(&arg);
4353     return createDeviceArgumentAccessor(mapData, arg, input, retVal, builder,
4373   for (auto [arg, var] : llvm::zip_equal(hostEvalBlockArgs, hostEvalVars)) {
4375     moduleTranslation.mapValue(arg, value);