19dfe4e7cSTobias Grosser //===------ PPCGCodeGeneration.cpp - Polly Accelerator Code Generation. ---===//
29dfe4e7cSTobias Grosser //
39dfe4e7cSTobias Grosser //                     The LLVM Compiler Infrastructure
49dfe4e7cSTobias Grosser //
59dfe4e7cSTobias Grosser // This file is distributed under the University of Illinois Open Source
69dfe4e7cSTobias Grosser // License. See LICENSE.TXT for details.
79dfe4e7cSTobias Grosser //
89dfe4e7cSTobias Grosser //===----------------------------------------------------------------------===//
99dfe4e7cSTobias Grosser //
109dfe4e7cSTobias Grosser // Take a scop created by ScopInfo and map it to GPU code using the ppcg
119dfe4e7cSTobias Grosser // GPU mapping strategy.
129dfe4e7cSTobias Grosser //
139dfe4e7cSTobias Grosser //===----------------------------------------------------------------------===//
149dfe4e7cSTobias Grosser 
159dfe4e7cSTobias Grosser #include "polly/CodeGen/IslNodeBuilder.h"
1638fc0aedSTobias Grosser #include "polly/CodeGen/Utils.h"
179dfe4e7cSTobias Grosser #include "polly/DependenceInfo.h"
189dfe4e7cSTobias Grosser #include "polly/LinkAllPasses.h"
19f384594dSTobias Grosser #include "polly/Options.h"
209dfe4e7cSTobias Grosser #include "polly/ScopInfo.h"
21edb885cbSTobias Grosser #include "polly/Support/SCEVValidator.h"
2274dc3cb4STobias Grosser #include "llvm/ADT/PostOrderIterator.h"
239dfe4e7cSTobias Grosser #include "llvm/Analysis/AliasAnalysis.h"
249dfe4e7cSTobias Grosser #include "llvm/Analysis/BasicAliasAnalysis.h"
259dfe4e7cSTobias Grosser #include "llvm/Analysis/GlobalsModRef.h"
269dfe4e7cSTobias Grosser #include "llvm/Analysis/PostDominators.h"
279dfe4e7cSTobias Grosser #include "llvm/Analysis/ScalarEvolutionAliasAnalysis.h"
2874dc3cb4STobias Grosser #include "llvm/Analysis/TargetLibraryInfo.h"
2974dc3cb4STobias Grosser #include "llvm/Analysis/TargetTransformInfo.h"
3074dc3cb4STobias Grosser #include "llvm/IR/LegacyPassManager.h"
31e1a98343STobias Grosser #include "llvm/IR/Verifier.h"
3274dc3cb4STobias Grosser #include "llvm/Support/TargetRegistry.h"
3374dc3cb4STobias Grosser #include "llvm/Support/TargetSelect.h"
3474dc3cb4STobias Grosser #include "llvm/Target/TargetMachine.h"
359a18d559STobias Grosser #include "llvm/Transforms/IPO/PassManagerBuilder.h"
369dfe4e7cSTobias Grosser 
37f384594dSTobias Grosser #include "isl/union_map.h"
38f384594dSTobias Grosser 
39e938517eSTobias Grosser extern "C" {
40a56f8f8eSTobias Grosser #include "ppcg/cuda.h"
41a56f8f8eSTobias Grosser #include "ppcg/gpu.h"
42a56f8f8eSTobias Grosser #include "ppcg/gpu_print.h"
43a56f8f8eSTobias Grosser #include "ppcg/ppcg.h"
44a56f8f8eSTobias Grosser #include "ppcg/schedule.h"
45e938517eSTobias Grosser }
46e938517eSTobias Grosser 
479dfe4e7cSTobias Grosser #include "llvm/Support/Debug.h"
489dfe4e7cSTobias Grosser 
499dfe4e7cSTobias Grosser using namespace polly;
509dfe4e7cSTobias Grosser using namespace llvm;
519dfe4e7cSTobias Grosser 
529dfe4e7cSTobias Grosser #define DEBUG_TYPE "polly-codegen-ppcg"
539dfe4e7cSTobias Grosser 
54f384594dSTobias Grosser static cl::opt<bool> DumpSchedule("polly-acc-dump-schedule",
55f384594dSTobias Grosser                                   cl::desc("Dump the computed GPU Schedule"),
56681bd568STobias Grosser                                   cl::Hidden, cl::init(false), cl::ZeroOrMore,
57f384594dSTobias Grosser                                   cl::cat(PollyCategory));
5869b46751STobias Grosser 
5969b46751STobias Grosser static cl::opt<bool>
6069b46751STobias Grosser     DumpCode("polly-acc-dump-code",
6169b46751STobias Grosser              cl::desc("Dump C code describing the GPU mapping"), cl::Hidden,
6269b46751STobias Grosser              cl::init(false), cl::ZeroOrMore, cl::cat(PollyCategory));
6369b46751STobias Grosser 
6432837fe3STobias Grosser static cl::opt<bool> DumpKernelIR("polly-acc-dump-kernel-ir",
6532837fe3STobias Grosser                                   cl::desc("Dump the kernel LLVM-IR"),
6632837fe3STobias Grosser                                   cl::Hidden, cl::init(false), cl::ZeroOrMore,
6732837fe3STobias Grosser                                   cl::cat(PollyCategory));
6832837fe3STobias Grosser 
6974dc3cb4STobias Grosser static cl::opt<bool> DumpKernelASM("polly-acc-dump-kernel-asm",
7074dc3cb4STobias Grosser                                    cl::desc("Dump the kernel assembly code"),
7174dc3cb4STobias Grosser                                    cl::Hidden, cl::init(false), cl::ZeroOrMore,
7274dc3cb4STobias Grosser                                    cl::cat(PollyCategory));
7374dc3cb4STobias Grosser 
7474dc3cb4STobias Grosser static cl::opt<bool> FastMath("polly-acc-fastmath",
7574dc3cb4STobias Grosser                               cl::desc("Allow unsafe math optimizations"),
7674dc3cb4STobias Grosser                               cl::Hidden, cl::init(false), cl::ZeroOrMore,
7774dc3cb4STobias Grosser                               cl::cat(PollyCategory));
7874dc3cb4STobias Grosser 
7974dc3cb4STobias Grosser static cl::opt<std::string>
8074dc3cb4STobias Grosser     CudaVersion("polly-acc-cuda-version",
8174dc3cb4STobias Grosser                 cl::desc("The CUDA version to compile for"), cl::Hidden,
8274dc3cb4STobias Grosser                 cl::init("sm_30"), cl::ZeroOrMore, cl::cat(PollyCategory));
8374dc3cb4STobias Grosser 
8460c60025STobias Grosser /// Create the ast expressions for a ScopStmt.
8560c60025STobias Grosser ///
8660c60025STobias Grosser /// This function is a callback for to generate the ast expressions for each
8760c60025STobias Grosser /// of the scheduled ScopStmts.
8860c60025STobias Grosser static __isl_give isl_id_to_ast_expr *pollyBuildAstExprForStmt(
89edb885cbSTobias Grosser     void *StmtT, isl_ast_build *Build,
9060c60025STobias Grosser     isl_multi_pw_aff *(*FunctionIndex)(__isl_take isl_multi_pw_aff *MPA,
9160c60025STobias Grosser                                        isl_id *Id, void *User),
9260c60025STobias Grosser     void *UserIndex,
9360c60025STobias Grosser     isl_ast_expr *(*FunctionExpr)(isl_ast_expr *Expr, isl_id *Id, void *User),
94edb885cbSTobias Grosser     void *UserExpr) {
9560c60025STobias Grosser 
96edb885cbSTobias Grosser   ScopStmt *Stmt = (ScopStmt *)StmtT;
9760c60025STobias Grosser 
98edb885cbSTobias Grosser   isl_ctx *Ctx;
99edb885cbSTobias Grosser 
100edb885cbSTobias Grosser   if (!Stmt || !Build)
101edb885cbSTobias Grosser     return NULL;
102edb885cbSTobias Grosser 
103edb885cbSTobias Grosser   Ctx = isl_ast_build_get_ctx(Build);
104edb885cbSTobias Grosser   isl_id_to_ast_expr *RefToExpr = isl_id_to_ast_expr_alloc(Ctx, 0);
105edb885cbSTobias Grosser 
106edb885cbSTobias Grosser   for (MemoryAccess *Acc : *Stmt) {
107edb885cbSTobias Grosser     isl_map *AddrFunc = Acc->getAddressFunction();
108edb885cbSTobias Grosser     AddrFunc = isl_map_intersect_domain(AddrFunc, Stmt->getDomain());
109edb885cbSTobias Grosser     isl_id *RefId = Acc->getId();
110edb885cbSTobias Grosser     isl_pw_multi_aff *PMA = isl_pw_multi_aff_from_map(AddrFunc);
111edb885cbSTobias Grosser     isl_multi_pw_aff *MPA = isl_multi_pw_aff_from_pw_multi_aff(PMA);
112edb885cbSTobias Grosser     MPA = isl_multi_pw_aff_coalesce(MPA);
113edb885cbSTobias Grosser     MPA = FunctionIndex(MPA, RefId, UserIndex);
114edb885cbSTobias Grosser     isl_ast_expr *Access = isl_ast_build_access_from_multi_pw_aff(Build, MPA);
115edb885cbSTobias Grosser     Access = FunctionExpr(Access, RefId, UserExpr);
116edb885cbSTobias Grosser     RefToExpr = isl_id_to_ast_expr_set(RefToExpr, RefId, Access);
117edb885cbSTobias Grosser   }
118edb885cbSTobias Grosser 
119edb885cbSTobias Grosser   return RefToExpr;
12060c60025STobias Grosser }
121f384594dSTobias Grosser 
12238fc0aedSTobias Grosser /// Generate code for a GPU specific isl AST.
12338fc0aedSTobias Grosser ///
12438fc0aedSTobias Grosser /// The GPUNodeBuilder augments the general existing IslNodeBuilder, which
12538fc0aedSTobias Grosser /// generates code for general-prupose AST nodes, with special functionality
12638fc0aedSTobias Grosser /// for generating GPU specific user nodes.
12738fc0aedSTobias Grosser ///
12838fc0aedSTobias Grosser /// @see GPUNodeBuilder::createUser
12938fc0aedSTobias Grosser class GPUNodeBuilder : public IslNodeBuilder {
13038fc0aedSTobias Grosser public:
13138fc0aedSTobias Grosser   GPUNodeBuilder(PollyIRBuilder &Builder, ScopAnnotator &Annotator, Pass *P,
13238fc0aedSTobias Grosser                  const DataLayout &DL, LoopInfo &LI, ScalarEvolution &SE,
13332837fe3STobias Grosser                  DominatorTree &DT, Scop &S, gpu_prog *Prog)
134edb885cbSTobias Grosser       : IslNodeBuilder(Builder, Annotator, P, DL, LI, SE, DT, S), Prog(Prog) {
135edb885cbSTobias Grosser     getExprBuilder().setIDToSAI(&IDToSAI);
136edb885cbSTobias Grosser   }
13738fc0aedSTobias Grosser 
138fa7b0802STobias Grosser   /// Create after-run-time-check initialization code.
139fa7b0802STobias Grosser   void initializeAfterRTH();
140fa7b0802STobias Grosser 
141fa7b0802STobias Grosser   /// Finalize the generated scop.
142fa7b0802STobias Grosser   virtual void finalize();
143fa7b0802STobias Grosser 
14438fc0aedSTobias Grosser private:
14574dc3cb4STobias Grosser   /// A vector of array base pointers for which a new ScopArrayInfo was created.
14674dc3cb4STobias Grosser   ///
14774dc3cb4STobias Grosser   /// This vector is used to delete the ScopArrayInfo when it is not needed any
14874dc3cb4STobias Grosser   /// more.
14974dc3cb4STobias Grosser   std::vector<Value *> LocalArrays;
15074dc3cb4STobias Grosser 
15113c78e4dSTobias Grosser   /// A map from ScopArrays to their corresponding device allocations.
15213c78e4dSTobias Grosser   std::map<ScopArrayInfo *, Value *> DeviceAllocations;
1537287aeddSTobias Grosser 
154fa7b0802STobias Grosser   /// The current GPU context.
155fa7b0802STobias Grosser   Value *GPUContext;
156fa7b0802STobias Grosser 
15732837fe3STobias Grosser   /// A module containing GPU code.
15832837fe3STobias Grosser   ///
15932837fe3STobias Grosser   /// This pointer is only set in case we are currently generating GPU code.
16032837fe3STobias Grosser   std::unique_ptr<Module> GPUModule;
16132837fe3STobias Grosser 
16232837fe3STobias Grosser   /// The GPU program we generate code for.
16332837fe3STobias Grosser   gpu_prog *Prog;
16432837fe3STobias Grosser 
165472f9654STobias Grosser   /// Class to free isl_ids.
166472f9654STobias Grosser   class IslIdDeleter {
167472f9654STobias Grosser   public:
168472f9654STobias Grosser     void operator()(__isl_take isl_id *Id) { isl_id_free(Id); };
169472f9654STobias Grosser   };
170472f9654STobias Grosser 
171472f9654STobias Grosser   /// A set containing all isl_ids allocated in a GPU kernel.
172472f9654STobias Grosser   ///
173472f9654STobias Grosser   /// By releasing this set all isl_ids will be freed.
174472f9654STobias Grosser   std::set<std::unique_ptr<isl_id, IslIdDeleter>> KernelIDs;
175472f9654STobias Grosser 
176edb885cbSTobias Grosser   IslExprBuilder::IDToScopArrayInfoTy IDToSAI;
177edb885cbSTobias Grosser 
17838fc0aedSTobias Grosser   /// Create code for user-defined AST nodes.
17938fc0aedSTobias Grosser   ///
18038fc0aedSTobias Grosser   /// These AST nodes can be of type:
18138fc0aedSTobias Grosser   ///
18238fc0aedSTobias Grosser   ///   - ScopStmt:      A computational statement (TODO)
18338fc0aedSTobias Grosser   ///   - Kernel:        A GPU kernel call (TODO)
18413c78e4dSTobias Grosser   ///   - Data-Transfer: A GPU <-> CPU data-transfer
1855260c041STobias Grosser   ///   - In-kernel synchronization
1865260c041STobias Grosser   ///   - In-kernel memory copy statement
18738fc0aedSTobias Grosser   ///
1881fb9b64dSTobias Grosser   /// @param UserStmt The ast node to generate code for.
1891fb9b64dSTobias Grosser   virtual void createUser(__isl_take isl_ast_node *UserStmt);
19032837fe3STobias Grosser 
19113c78e4dSTobias Grosser   enum DataDirection { HOST_TO_DEVICE, DEVICE_TO_HOST };
19213c78e4dSTobias Grosser 
19313c78e4dSTobias Grosser   /// Create code for a data transfer statement
19413c78e4dSTobias Grosser   ///
19513c78e4dSTobias Grosser   /// @param TransferStmt The data transfer statement.
19613c78e4dSTobias Grosser   /// @param Direction The direction in which to transfer data.
19713c78e4dSTobias Grosser   void createDataTransfer(__isl_take isl_ast_node *TransferStmt,
19813c78e4dSTobias Grosser                           enum DataDirection Direction);
19913c78e4dSTobias Grosser 
200edb885cbSTobias Grosser   /// Find llvm::Values referenced in GPU kernel.
201edb885cbSTobias Grosser   ///
202edb885cbSTobias Grosser   /// @param Kernel The kernel to scan for llvm::Values
203edb885cbSTobias Grosser   ///
204edb885cbSTobias Grosser   /// @returns A set of values referenced by the kernel.
205edb885cbSTobias Grosser   SetVector<Value *> getReferencesInKernel(ppcg_kernel *Kernel);
206edb885cbSTobias Grosser 
20779a947c2STobias Grosser   /// Compute the sizes of the execution grid for a given kernel.
20879a947c2STobias Grosser   ///
20979a947c2STobias Grosser   /// @param Kernel The kernel to compute grid sizes for.
21079a947c2STobias Grosser   ///
21179a947c2STobias Grosser   /// @returns A tuple with grid sizes for X and Y dimension
21279a947c2STobias Grosser   std::tuple<Value *, Value *> getGridSizes(ppcg_kernel *Kernel);
21379a947c2STobias Grosser 
21479a947c2STobias Grosser   /// Compute the sizes of the thread blocks for a given kernel.
21579a947c2STobias Grosser   ///
21679a947c2STobias Grosser   /// @param Kernel The kernel to compute thread block sizes for.
21779a947c2STobias Grosser   ///
21879a947c2STobias Grosser   /// @returns A tuple with thread block sizes for X, Y, and Z dimensions.
21979a947c2STobias Grosser   std::tuple<Value *, Value *, Value *> getBlockSizes(ppcg_kernel *Kernel);
22079a947c2STobias Grosser 
22179a947c2STobias Grosser   /// Create kernel launch parameters.
22279a947c2STobias Grosser   ///
22379a947c2STobias Grosser   /// @param Kernel The kernel to create parameters for.
22479a947c2STobias Grosser   /// @param F      The kernel function that has been created.
22579a947c2STobias Grosser   ///
22679a947c2STobias Grosser   /// @returns A stack allocated array with pointers to the parameter
22779a947c2STobias Grosser   ///          values that are passed to the kernel.
22879a947c2STobias Grosser   Value *createLaunchParameters(ppcg_kernel *Kernel, Function *F);
22979a947c2STobias Grosser 
23032837fe3STobias Grosser   /// Create GPU kernel.
23132837fe3STobias Grosser   ///
23232837fe3STobias Grosser   /// Code generate the kernel described by @p KernelStmt.
23332837fe3STobias Grosser   ///
23432837fe3STobias Grosser   /// @param KernelStmt The ast node to generate kernel code for.
23532837fe3STobias Grosser   void createKernel(__isl_take isl_ast_node *KernelStmt);
23632837fe3STobias Grosser 
23713c78e4dSTobias Grosser   /// Generate code that computes the size of an array.
23813c78e4dSTobias Grosser   ///
23913c78e4dSTobias Grosser   /// @param Array The array for which to compute a size.
24013c78e4dSTobias Grosser   Value *getArraySize(gpu_array_info *Array);
24113c78e4dSTobias Grosser 
24232837fe3STobias Grosser   /// Create kernel function.
24332837fe3STobias Grosser   ///
24432837fe3STobias Grosser   /// Create a kernel function located in a newly created module that can serve
24532837fe3STobias Grosser   /// as target for device code generation. Set the Builder to point to the
24632837fe3STobias Grosser   /// start block of this newly created function.
24732837fe3STobias Grosser   ///
24832837fe3STobias Grosser   /// @param Kernel The kernel to generate code for.
249edb885cbSTobias Grosser   /// @param SubtreeValues The set of llvm::Values referenced by this kernel.
250edb885cbSTobias Grosser   void createKernelFunction(ppcg_kernel *Kernel,
251edb885cbSTobias Grosser                             SetVector<Value *> &SubtreeValues);
25232837fe3STobias Grosser 
25332837fe3STobias Grosser   /// Create the declaration of a kernel function.
25432837fe3STobias Grosser   ///
25532837fe3STobias Grosser   /// The kernel function takes as arguments:
25632837fe3STobias Grosser   ///
25732837fe3STobias Grosser   ///   - One i8 pointer for each external array reference used in the kernel.
258f6044bd0STobias Grosser   ///   - Host iterators
259c84a1995STobias Grosser   ///   - Parameters
26032837fe3STobias Grosser   ///   - Other LLVM Value references (TODO)
26132837fe3STobias Grosser   ///
26232837fe3STobias Grosser   /// @param Kernel The kernel to generate the function declaration for.
263edb885cbSTobias Grosser   /// @param SubtreeValues The set of llvm::Values referenced by this kernel.
264edb885cbSTobias Grosser   ///
26532837fe3STobias Grosser   /// @returns The newly declared function.
266edb885cbSTobias Grosser   Function *createKernelFunctionDecl(ppcg_kernel *Kernel,
267edb885cbSTobias Grosser                                      SetVector<Value *> &SubtreeValues);
26832837fe3STobias Grosser 
269472f9654STobias Grosser   /// Insert intrinsic functions to obtain thread and block ids.
270472f9654STobias Grosser   ///
271472f9654STobias Grosser   /// @param The kernel to generate the intrinsic functions for.
272472f9654STobias Grosser   void insertKernelIntrinsics(ppcg_kernel *Kernel);
273472f9654STobias Grosser 
274edb885cbSTobias Grosser   /// Create code for a ScopStmt called in @p Expr.
275edb885cbSTobias Grosser   ///
276edb885cbSTobias Grosser   /// @param Expr The expression containing the call.
277edb885cbSTobias Grosser   /// @param KernelStmt The kernel statement referenced in the call.
278edb885cbSTobias Grosser   void createScopStmt(isl_ast_expr *Expr, ppcg_kernel_stmt *KernelStmt);
279edb885cbSTobias Grosser 
2805260c041STobias Grosser   /// Create an in-kernel synchronization call.
2815260c041STobias Grosser   void createKernelSync();
2825260c041STobias Grosser 
28374dc3cb4STobias Grosser   /// Create a PTX assembly string for the current GPU kernel.
28474dc3cb4STobias Grosser   ///
28574dc3cb4STobias Grosser   /// @returns A string containing the corresponding PTX assembly code.
28674dc3cb4STobias Grosser   std::string createKernelASM();
28774dc3cb4STobias Grosser 
28874dc3cb4STobias Grosser   /// Remove references from the dominator tree to the kernel function @p F.
28974dc3cb4STobias Grosser   ///
29074dc3cb4STobias Grosser   /// @param F The function to remove references to.
29174dc3cb4STobias Grosser   void clearDominators(Function *F);
29274dc3cb4STobias Grosser 
29374dc3cb4STobias Grosser   /// Remove references from scalar evolution to the kernel function @p F.
29474dc3cb4STobias Grosser   ///
29574dc3cb4STobias Grosser   /// @param F The function to remove references to.
29674dc3cb4STobias Grosser   void clearScalarEvolution(Function *F);
29774dc3cb4STobias Grosser 
29874dc3cb4STobias Grosser   /// Remove references from loop info to the kernel function @p F.
29974dc3cb4STobias Grosser   ///
30074dc3cb4STobias Grosser   /// @param F The function to remove references to.
30174dc3cb4STobias Grosser   void clearLoops(Function *F);
30274dc3cb4STobias Grosser 
30332837fe3STobias Grosser   /// Finalize the generation of the kernel function.
30432837fe3STobias Grosser   ///
30532837fe3STobias Grosser   /// Free the LLVM-IR module corresponding to the kernel and -- if requested --
30632837fe3STobias Grosser   /// dump its IR to stderr.
30757793596STobias Grosser   ///
30857793596STobias Grosser   /// @returns The Assembly string of the kernel.
30957793596STobias Grosser   std::string finalizeKernelFunction();
310fa7b0802STobias Grosser 
3117287aeddSTobias Grosser   /// Create code that allocates memory to store arrays on device.
312fa7b0802STobias Grosser   void allocateDeviceArrays();
313fa7b0802STobias Grosser 
3147287aeddSTobias Grosser   /// Free all allocated device arrays.
3157287aeddSTobias Grosser   void freeDeviceArrays();
3167287aeddSTobias Grosser 
317fa7b0802STobias Grosser   /// Create a call to initialize the GPU context.
318fa7b0802STobias Grosser   ///
319fa7b0802STobias Grosser   /// @returns A pointer to the newly initialized context.
320fa7b0802STobias Grosser   Value *createCallInitContext();
321fa7b0802STobias Grosser 
32279a947c2STobias Grosser   /// Create a call to get the device pointer for a kernel allocation.
32379a947c2STobias Grosser   ///
32479a947c2STobias Grosser   /// @param Allocation The Polly GPU allocation
32579a947c2STobias Grosser   ///
32679a947c2STobias Grosser   /// @returns The device parameter corresponding to this allocation.
32779a947c2STobias Grosser   Value *createCallGetDevicePtr(Value *Allocation);
32879a947c2STobias Grosser 
329fa7b0802STobias Grosser   /// Create a call to free the GPU context.
330fa7b0802STobias Grosser   ///
331fa7b0802STobias Grosser   /// @param Context A pointer to an initialized GPU context.
332fa7b0802STobias Grosser   void createCallFreeContext(Value *Context);
333fa7b0802STobias Grosser 
3347287aeddSTobias Grosser   /// Create a call to allocate memory on the device.
3357287aeddSTobias Grosser   ///
3367287aeddSTobias Grosser   /// @param Size The size of memory to allocate
3377287aeddSTobias Grosser   ///
3387287aeddSTobias Grosser   /// @returns A pointer that identifies this allocation.
339fa7b0802STobias Grosser   Value *createCallAllocateMemoryForDevice(Value *Size);
3407287aeddSTobias Grosser 
3417287aeddSTobias Grosser   /// Create a call to free a device array.
3427287aeddSTobias Grosser   ///
3437287aeddSTobias Grosser   /// @param Array The device array to free.
3447287aeddSTobias Grosser   void createCallFreeDeviceMemory(Value *Array);
34513c78e4dSTobias Grosser 
34613c78e4dSTobias Grosser   /// Create a call to copy data from host to device.
34713c78e4dSTobias Grosser   ///
34813c78e4dSTobias Grosser   /// @param HostPtr A pointer to the host data that should be copied.
34913c78e4dSTobias Grosser   /// @param DevicePtr A device pointer specifying the location to copy to.
35013c78e4dSTobias Grosser   void createCallCopyFromHostToDevice(Value *HostPtr, Value *DevicePtr,
35113c78e4dSTobias Grosser                                       Value *Size);
35213c78e4dSTobias Grosser 
35313c78e4dSTobias Grosser   /// Create a call to copy data from device to host.
35413c78e4dSTobias Grosser   ///
35513c78e4dSTobias Grosser   /// @param DevicePtr A pointer to the device data that should be copied.
35613c78e4dSTobias Grosser   /// @param HostPtr A host pointer specifying the location to copy to.
35713c78e4dSTobias Grosser   void createCallCopyFromDeviceToHost(Value *DevicePtr, Value *HostPtr,
35813c78e4dSTobias Grosser                                       Value *Size);
35957793596STobias Grosser 
36057793596STobias Grosser   /// Create a call to get a kernel from an assembly string.
36157793596STobias Grosser   ///
36257793596STobias Grosser   /// @param Buffer The string describing the kernel.
36357793596STobias Grosser   /// @param Entry  The name of the kernel function to call.
36457793596STobias Grosser   ///
36557793596STobias Grosser   /// @returns A pointer to a kernel object
36657793596STobias Grosser   Value *createCallGetKernel(Value *Buffer, Value *Entry);
36757793596STobias Grosser 
36857793596STobias Grosser   /// Create a call to free a GPU kernel.
36957793596STobias Grosser   ///
37057793596STobias Grosser   /// @param GPUKernel THe kernel to free.
37157793596STobias Grosser   void createCallFreeKernel(Value *GPUKernel);
37279a947c2STobias Grosser 
37379a947c2STobias Grosser   /// Create a call to launch a GPU kernel.
37479a947c2STobias Grosser   ///
37579a947c2STobias Grosser   /// @param GPUKernel  The kernel to launch.
37679a947c2STobias Grosser   /// @param GridDimX   The size of the first grid dimension.
37779a947c2STobias Grosser   /// @param GridDimY   The size of the second grid dimension.
37879a947c2STobias Grosser   /// @param GridBlockX The size of the first block dimension.
37979a947c2STobias Grosser   /// @param GridBlockY The size of the second block dimension.
38079a947c2STobias Grosser   /// @param GridBlockZ The size of the third block dimension.
38179a947c2STobias Grosser   /// @param Paramters  A pointer to an array that contains itself pointers to
38279a947c2STobias Grosser   ///                   the parameter values passed for each kernel argument.
38379a947c2STobias Grosser   void createCallLaunchKernel(Value *GPUKernel, Value *GridDimX,
38479a947c2STobias Grosser                               Value *GridDimY, Value *BlockDimX,
38579a947c2STobias Grosser                               Value *BlockDimY, Value *BlockDimZ,
38679a947c2STobias Grosser                               Value *Parameters);
3871fb9b64dSTobias Grosser };
3881fb9b64dSTobias Grosser 
389fa7b0802STobias Grosser void GPUNodeBuilder::initializeAfterRTH() {
390fa7b0802STobias Grosser   GPUContext = createCallInitContext();
391fa7b0802STobias Grosser   allocateDeviceArrays();
392fa7b0802STobias Grosser }
393fa7b0802STobias Grosser 
394fa7b0802STobias Grosser void GPUNodeBuilder::finalize() {
3957287aeddSTobias Grosser   freeDeviceArrays();
396fa7b0802STobias Grosser   createCallFreeContext(GPUContext);
397fa7b0802STobias Grosser   IslNodeBuilder::finalize();
398fa7b0802STobias Grosser }
399fa7b0802STobias Grosser 
400fa7b0802STobias Grosser void GPUNodeBuilder::allocateDeviceArrays() {
401fa7b0802STobias Grosser   isl_ast_build *Build = isl_ast_build_from_context(S.getContext());
402fa7b0802STobias Grosser 
403fa7b0802STobias Grosser   for (int i = 0; i < Prog->n_array; ++i) {
404fa7b0802STobias Grosser     gpu_array_info *Array = &Prog->array[i];
40513c78e4dSTobias Grosser     auto *ScopArray = (ScopArrayInfo *)Array->user;
4067287aeddSTobias Grosser     std::string DevArrayName("p_dev_array_");
4077287aeddSTobias Grosser     DevArrayName.append(Array->name);
408fa7b0802STobias Grosser 
40913c78e4dSTobias Grosser     Value *ArraySize = getArraySize(Array);
4107287aeddSTobias Grosser     Value *DevArray = createCallAllocateMemoryForDevice(ArraySize);
4117287aeddSTobias Grosser     DevArray->setName(DevArrayName);
41213c78e4dSTobias Grosser     DeviceAllocations[ScopArray] = DevArray;
413fa7b0802STobias Grosser   }
414fa7b0802STobias Grosser 
415fa7b0802STobias Grosser   isl_ast_build_free(Build);
416fa7b0802STobias Grosser }
417fa7b0802STobias Grosser 
4187287aeddSTobias Grosser void GPUNodeBuilder::freeDeviceArrays() {
41913c78e4dSTobias Grosser   for (auto &Array : DeviceAllocations)
42013c78e4dSTobias Grosser     createCallFreeDeviceMemory(Array.second);
4217287aeddSTobias Grosser }
4227287aeddSTobias Grosser 
42357793596STobias Grosser Value *GPUNodeBuilder::createCallGetKernel(Value *Buffer, Value *Entry) {
42457793596STobias Grosser   const char *Name = "polly_getKernel";
42557793596STobias Grosser   Module *M = Builder.GetInsertBlock()->getParent()->getParent();
42657793596STobias Grosser   Function *F = M->getFunction(Name);
42757793596STobias Grosser 
42857793596STobias Grosser   // If F is not available, declare it.
42957793596STobias Grosser   if (!F) {
43057793596STobias Grosser     GlobalValue::LinkageTypes Linkage = Function::ExternalLinkage;
43157793596STobias Grosser     std::vector<Type *> Args;
43257793596STobias Grosser     Args.push_back(Builder.getInt8PtrTy());
43357793596STobias Grosser     Args.push_back(Builder.getInt8PtrTy());
43457793596STobias Grosser     FunctionType *Ty = FunctionType::get(Builder.getInt8PtrTy(), Args, false);
43557793596STobias Grosser     F = Function::Create(Ty, Linkage, Name, M);
43657793596STobias Grosser   }
43757793596STobias Grosser 
43857793596STobias Grosser   return Builder.CreateCall(F, {Buffer, Entry});
43957793596STobias Grosser }
44057793596STobias Grosser 
44179a947c2STobias Grosser Value *GPUNodeBuilder::createCallGetDevicePtr(Value *Allocation) {
44279a947c2STobias Grosser   const char *Name = "polly_getDevicePtr";
44379a947c2STobias Grosser   Module *M = Builder.GetInsertBlock()->getParent()->getParent();
44479a947c2STobias Grosser   Function *F = M->getFunction(Name);
44579a947c2STobias Grosser 
44679a947c2STobias Grosser   // If F is not available, declare it.
44779a947c2STobias Grosser   if (!F) {
44879a947c2STobias Grosser     GlobalValue::LinkageTypes Linkage = Function::ExternalLinkage;
44979a947c2STobias Grosser     std::vector<Type *> Args;
45079a947c2STobias Grosser     Args.push_back(Builder.getInt8PtrTy());
45179a947c2STobias Grosser     FunctionType *Ty = FunctionType::get(Builder.getInt8PtrTy(), Args, false);
45279a947c2STobias Grosser     F = Function::Create(Ty, Linkage, Name, M);
45379a947c2STobias Grosser   }
45479a947c2STobias Grosser 
45579a947c2STobias Grosser   return Builder.CreateCall(F, {Allocation});
45679a947c2STobias Grosser }
45779a947c2STobias Grosser 
45879a947c2STobias Grosser void GPUNodeBuilder::createCallLaunchKernel(Value *GPUKernel, Value *GridDimX,
45979a947c2STobias Grosser                                             Value *GridDimY, Value *BlockDimX,
46079a947c2STobias Grosser                                             Value *BlockDimY, Value *BlockDimZ,
46179a947c2STobias Grosser                                             Value *Parameters) {
46279a947c2STobias Grosser   const char *Name = "polly_launchKernel";
46379a947c2STobias Grosser   Module *M = Builder.GetInsertBlock()->getParent()->getParent();
46479a947c2STobias Grosser   Function *F = M->getFunction(Name);
46579a947c2STobias Grosser 
46679a947c2STobias Grosser   // If F is not available, declare it.
46779a947c2STobias Grosser   if (!F) {
46879a947c2STobias Grosser     GlobalValue::LinkageTypes Linkage = Function::ExternalLinkage;
46979a947c2STobias Grosser     std::vector<Type *> Args;
47079a947c2STobias Grosser     Args.push_back(Builder.getInt8PtrTy());
47179a947c2STobias Grosser     Args.push_back(Builder.getInt32Ty());
47279a947c2STobias Grosser     Args.push_back(Builder.getInt32Ty());
47379a947c2STobias Grosser     Args.push_back(Builder.getInt32Ty());
47479a947c2STobias Grosser     Args.push_back(Builder.getInt32Ty());
47579a947c2STobias Grosser     Args.push_back(Builder.getInt32Ty());
47679a947c2STobias Grosser     Args.push_back(Builder.getInt8PtrTy());
47779a947c2STobias Grosser     FunctionType *Ty = FunctionType::get(Builder.getVoidTy(), Args, false);
47879a947c2STobias Grosser     F = Function::Create(Ty, Linkage, Name, M);
47979a947c2STobias Grosser   }
48079a947c2STobias Grosser 
48179a947c2STobias Grosser   Builder.CreateCall(F, {GPUKernel, GridDimX, GridDimY, BlockDimX, BlockDimY,
48279a947c2STobias Grosser                          BlockDimZ, Parameters});
48379a947c2STobias Grosser }
48479a947c2STobias Grosser 
48557793596STobias Grosser void GPUNodeBuilder::createCallFreeKernel(Value *GPUKernel) {
48657793596STobias Grosser   const char *Name = "polly_freeKernel";
48757793596STobias Grosser   Module *M = Builder.GetInsertBlock()->getParent()->getParent();
48857793596STobias Grosser   Function *F = M->getFunction(Name);
48957793596STobias Grosser 
49057793596STobias Grosser   // If F is not available, declare it.
49157793596STobias Grosser   if (!F) {
49257793596STobias Grosser     GlobalValue::LinkageTypes Linkage = Function::ExternalLinkage;
49357793596STobias Grosser     std::vector<Type *> Args;
49457793596STobias Grosser     Args.push_back(Builder.getInt8PtrTy());
49557793596STobias Grosser     FunctionType *Ty = FunctionType::get(Builder.getVoidTy(), Args, false);
49657793596STobias Grosser     F = Function::Create(Ty, Linkage, Name, M);
49757793596STobias Grosser   }
49857793596STobias Grosser 
49957793596STobias Grosser   Builder.CreateCall(F, {GPUKernel});
50057793596STobias Grosser }
50157793596STobias Grosser 
5027287aeddSTobias Grosser void GPUNodeBuilder::createCallFreeDeviceMemory(Value *Array) {
5037287aeddSTobias Grosser   const char *Name = "polly_freeDeviceMemory";
5047287aeddSTobias Grosser   Module *M = Builder.GetInsertBlock()->getParent()->getParent();
5057287aeddSTobias Grosser   Function *F = M->getFunction(Name);
5067287aeddSTobias Grosser 
5077287aeddSTobias Grosser   // If F is not available, declare it.
5087287aeddSTobias Grosser   if (!F) {
5097287aeddSTobias Grosser     GlobalValue::LinkageTypes Linkage = Function::ExternalLinkage;
5107287aeddSTobias Grosser     std::vector<Type *> Args;
5117287aeddSTobias Grosser     Args.push_back(Builder.getInt8PtrTy());
5127287aeddSTobias Grosser     FunctionType *Ty = FunctionType::get(Builder.getVoidTy(), Args, false);
5137287aeddSTobias Grosser     F = Function::Create(Ty, Linkage, Name, M);
5147287aeddSTobias Grosser   }
5157287aeddSTobias Grosser 
5167287aeddSTobias Grosser   Builder.CreateCall(F, {Array});
5177287aeddSTobias Grosser }
5187287aeddSTobias Grosser 
519fa7b0802STobias Grosser Value *GPUNodeBuilder::createCallAllocateMemoryForDevice(Value *Size) {
520fa7b0802STobias Grosser   const char *Name = "polly_allocateMemoryForDevice";
521fa7b0802STobias Grosser   Module *M = Builder.GetInsertBlock()->getParent()->getParent();
522fa7b0802STobias Grosser   Function *F = M->getFunction(Name);
523fa7b0802STobias Grosser 
524fa7b0802STobias Grosser   // If F is not available, declare it.
525fa7b0802STobias Grosser   if (!F) {
526fa7b0802STobias Grosser     GlobalValue::LinkageTypes Linkage = Function::ExternalLinkage;
527fa7b0802STobias Grosser     std::vector<Type *> Args;
528fa7b0802STobias Grosser     Args.push_back(Builder.getInt64Ty());
529fa7b0802STobias Grosser     FunctionType *Ty = FunctionType::get(Builder.getInt8PtrTy(), Args, false);
530fa7b0802STobias Grosser     F = Function::Create(Ty, Linkage, Name, M);
531fa7b0802STobias Grosser   }
532fa7b0802STobias Grosser 
533fa7b0802STobias Grosser   return Builder.CreateCall(F, {Size});
534fa7b0802STobias Grosser }
535fa7b0802STobias Grosser 
53613c78e4dSTobias Grosser void GPUNodeBuilder::createCallCopyFromHostToDevice(Value *HostData,
53713c78e4dSTobias Grosser                                                     Value *DeviceData,
53813c78e4dSTobias Grosser                                                     Value *Size) {
53913c78e4dSTobias Grosser   const char *Name = "polly_copyFromHostToDevice";
54013c78e4dSTobias Grosser   Module *M = Builder.GetInsertBlock()->getParent()->getParent();
54113c78e4dSTobias Grosser   Function *F = M->getFunction(Name);
54213c78e4dSTobias Grosser 
54313c78e4dSTobias Grosser   // If F is not available, declare it.
54413c78e4dSTobias Grosser   if (!F) {
54513c78e4dSTobias Grosser     GlobalValue::LinkageTypes Linkage = Function::ExternalLinkage;
54613c78e4dSTobias Grosser     std::vector<Type *> Args;
54713c78e4dSTobias Grosser     Args.push_back(Builder.getInt8PtrTy());
54813c78e4dSTobias Grosser     Args.push_back(Builder.getInt8PtrTy());
54913c78e4dSTobias Grosser     Args.push_back(Builder.getInt64Ty());
55013c78e4dSTobias Grosser     FunctionType *Ty = FunctionType::get(Builder.getVoidTy(), Args, false);
55113c78e4dSTobias Grosser     F = Function::Create(Ty, Linkage, Name, M);
55213c78e4dSTobias Grosser   }
55313c78e4dSTobias Grosser 
55413c78e4dSTobias Grosser   Builder.CreateCall(F, {HostData, DeviceData, Size});
55513c78e4dSTobias Grosser }
55613c78e4dSTobias Grosser 
55713c78e4dSTobias Grosser void GPUNodeBuilder::createCallCopyFromDeviceToHost(Value *DeviceData,
55813c78e4dSTobias Grosser                                                     Value *HostData,
55913c78e4dSTobias Grosser                                                     Value *Size) {
56013c78e4dSTobias Grosser   const char *Name = "polly_copyFromDeviceToHost";
56113c78e4dSTobias Grosser   Module *M = Builder.GetInsertBlock()->getParent()->getParent();
56213c78e4dSTobias Grosser   Function *F = M->getFunction(Name);
56313c78e4dSTobias Grosser 
56413c78e4dSTobias Grosser   // If F is not available, declare it.
56513c78e4dSTobias Grosser   if (!F) {
56613c78e4dSTobias Grosser     GlobalValue::LinkageTypes Linkage = Function::ExternalLinkage;
56713c78e4dSTobias Grosser     std::vector<Type *> Args;
56813c78e4dSTobias Grosser     Args.push_back(Builder.getInt8PtrTy());
56913c78e4dSTobias Grosser     Args.push_back(Builder.getInt8PtrTy());
57013c78e4dSTobias Grosser     Args.push_back(Builder.getInt64Ty());
57113c78e4dSTobias Grosser     FunctionType *Ty = FunctionType::get(Builder.getVoidTy(), Args, false);
57213c78e4dSTobias Grosser     F = Function::Create(Ty, Linkage, Name, M);
57313c78e4dSTobias Grosser   }
57413c78e4dSTobias Grosser 
57513c78e4dSTobias Grosser   Builder.CreateCall(F, {DeviceData, HostData, Size});
57613c78e4dSTobias Grosser }
57713c78e4dSTobias Grosser 
578fa7b0802STobias Grosser Value *GPUNodeBuilder::createCallInitContext() {
579fa7b0802STobias Grosser   const char *Name = "polly_initContext";
580fa7b0802STobias Grosser   Module *M = Builder.GetInsertBlock()->getParent()->getParent();
581fa7b0802STobias Grosser   Function *F = M->getFunction(Name);
582fa7b0802STobias Grosser 
583fa7b0802STobias Grosser   // If F is not available, declare it.
584fa7b0802STobias Grosser   if (!F) {
585fa7b0802STobias Grosser     GlobalValue::LinkageTypes Linkage = Function::ExternalLinkage;
586fa7b0802STobias Grosser     std::vector<Type *> Args;
587fa7b0802STobias Grosser     FunctionType *Ty = FunctionType::get(Builder.getInt8PtrTy(), Args, false);
588fa7b0802STobias Grosser     F = Function::Create(Ty, Linkage, Name, M);
589fa7b0802STobias Grosser   }
590fa7b0802STobias Grosser 
591fa7b0802STobias Grosser   return Builder.CreateCall(F, {});
592fa7b0802STobias Grosser }
593fa7b0802STobias Grosser 
594fa7b0802STobias Grosser void GPUNodeBuilder::createCallFreeContext(Value *Context) {
595fa7b0802STobias Grosser   const char *Name = "polly_freeContext";
596fa7b0802STobias Grosser   Module *M = Builder.GetInsertBlock()->getParent()->getParent();
597fa7b0802STobias Grosser   Function *F = M->getFunction(Name);
598fa7b0802STobias Grosser 
599fa7b0802STobias Grosser   // If F is not available, declare it.
600fa7b0802STobias Grosser   if (!F) {
601fa7b0802STobias Grosser     GlobalValue::LinkageTypes Linkage = Function::ExternalLinkage;
602fa7b0802STobias Grosser     std::vector<Type *> Args;
603fa7b0802STobias Grosser     Args.push_back(Builder.getInt8PtrTy());
604fa7b0802STobias Grosser     FunctionType *Ty = FunctionType::get(Builder.getVoidTy(), Args, false);
605fa7b0802STobias Grosser     F = Function::Create(Ty, Linkage, Name, M);
606fa7b0802STobias Grosser   }
607fa7b0802STobias Grosser 
608fa7b0802STobias Grosser   Builder.CreateCall(F, {Context});
609fa7b0802STobias Grosser }
610fa7b0802STobias Grosser 
6115260c041STobias Grosser /// Check if one string is a prefix of another.
6125260c041STobias Grosser ///
6135260c041STobias Grosser /// @param String The string in which to look for the prefix.
6145260c041STobias Grosser /// @param Prefix The prefix to look for.
6155260c041STobias Grosser static bool isPrefix(std::string String, std::string Prefix) {
6165260c041STobias Grosser   return String.find(Prefix) == 0;
6175260c041STobias Grosser }
6185260c041STobias Grosser 
61913c78e4dSTobias Grosser Value *GPUNodeBuilder::getArraySize(gpu_array_info *Array) {
62013c78e4dSTobias Grosser   isl_ast_build *Build = isl_ast_build_from_context(S.getContext());
62113c78e4dSTobias Grosser   Value *ArraySize = ConstantInt::get(Builder.getInt64Ty(), Array->size);
62213c78e4dSTobias Grosser 
62313c78e4dSTobias Grosser   if (!gpu_array_is_scalar(Array)) {
62413c78e4dSTobias Grosser     auto OffsetDimZero = isl_pw_aff_copy(Array->bound[0]);
62513c78e4dSTobias Grosser     isl_ast_expr *Res = isl_ast_build_expr_from_pw_aff(Build, OffsetDimZero);
62613c78e4dSTobias Grosser 
62713c78e4dSTobias Grosser     for (unsigned int i = 1; i < Array->n_index; i++) {
62813c78e4dSTobias Grosser       isl_pw_aff *Bound_I = isl_pw_aff_copy(Array->bound[i]);
62913c78e4dSTobias Grosser       isl_ast_expr *Expr = isl_ast_build_expr_from_pw_aff(Build, Bound_I);
63013c78e4dSTobias Grosser       Res = isl_ast_expr_mul(Res, Expr);
63113c78e4dSTobias Grosser     }
63213c78e4dSTobias Grosser 
63313c78e4dSTobias Grosser     Value *NumElements = ExprBuilder.create(Res);
63413c78e4dSTobias Grosser     ArraySize = Builder.CreateMul(ArraySize, NumElements);
63513c78e4dSTobias Grosser   }
63613c78e4dSTobias Grosser   isl_ast_build_free(Build);
63713c78e4dSTobias Grosser   return ArraySize;
63813c78e4dSTobias Grosser }
63913c78e4dSTobias Grosser 
64013c78e4dSTobias Grosser void GPUNodeBuilder::createDataTransfer(__isl_take isl_ast_node *TransferStmt,
64113c78e4dSTobias Grosser                                         enum DataDirection Direction) {
64213c78e4dSTobias Grosser   isl_ast_expr *Expr = isl_ast_node_user_get_expr(TransferStmt);
64313c78e4dSTobias Grosser   isl_ast_expr *Arg = isl_ast_expr_get_op_arg(Expr, 0);
64413c78e4dSTobias Grosser   isl_id *Id = isl_ast_expr_get_id(Arg);
64513c78e4dSTobias Grosser   auto Array = (gpu_array_info *)isl_id_get_user(Id);
64613c78e4dSTobias Grosser   auto ScopArray = (ScopArrayInfo *)(Array->user);
64713c78e4dSTobias Grosser 
64813c78e4dSTobias Grosser   Value *Size = getArraySize(Array);
64913c78e4dSTobias Grosser   Value *HostPtr = ScopArray->getBasePtr();
65013c78e4dSTobias Grosser 
65113c78e4dSTobias Grosser   Value *DevPtr = DeviceAllocations[ScopArray];
65213c78e4dSTobias Grosser 
65313c78e4dSTobias Grosser   if (gpu_array_is_scalar(Array)) {
65413c78e4dSTobias Grosser     HostPtr = Builder.CreateAlloca(ScopArray->getElementType());
65513c78e4dSTobias Grosser     Builder.CreateStore(ScopArray->getBasePtr(), HostPtr);
65613c78e4dSTobias Grosser   }
65713c78e4dSTobias Grosser 
65813c78e4dSTobias Grosser   HostPtr = Builder.CreatePointerCast(HostPtr, Builder.getInt8PtrTy());
65913c78e4dSTobias Grosser 
66013c78e4dSTobias Grosser   if (Direction == HOST_TO_DEVICE)
66113c78e4dSTobias Grosser     createCallCopyFromHostToDevice(HostPtr, DevPtr, Size);
66213c78e4dSTobias Grosser   else
66313c78e4dSTobias Grosser     createCallCopyFromDeviceToHost(DevPtr, HostPtr, Size);
66413c78e4dSTobias Grosser 
66513c78e4dSTobias Grosser   isl_id_free(Id);
66613c78e4dSTobias Grosser   isl_ast_expr_free(Arg);
66713c78e4dSTobias Grosser   isl_ast_expr_free(Expr);
66813c78e4dSTobias Grosser   isl_ast_node_free(TransferStmt);
66913c78e4dSTobias Grosser }
67013c78e4dSTobias Grosser 
6711fb9b64dSTobias Grosser void GPUNodeBuilder::createUser(__isl_take isl_ast_node *UserStmt) {
67232837fe3STobias Grosser   isl_ast_expr *Expr = isl_ast_node_user_get_expr(UserStmt);
67332837fe3STobias Grosser   isl_ast_expr *StmtExpr = isl_ast_expr_get_op_arg(Expr, 0);
67432837fe3STobias Grosser   isl_id *Id = isl_ast_expr_get_id(StmtExpr);
67532837fe3STobias Grosser   isl_id_free(Id);
67632837fe3STobias Grosser   isl_ast_expr_free(StmtExpr);
67732837fe3STobias Grosser 
67832837fe3STobias Grosser   const char *Str = isl_id_get_name(Id);
67932837fe3STobias Grosser   if (!strcmp(Str, "kernel")) {
68032837fe3STobias Grosser     createKernel(UserStmt);
68132837fe3STobias Grosser     isl_ast_expr_free(Expr);
68232837fe3STobias Grosser     return;
68332837fe3STobias Grosser   }
68432837fe3STobias Grosser 
68513c78e4dSTobias Grosser   if (isPrefix(Str, "to_device")) {
68613c78e4dSTobias Grosser     createDataTransfer(UserStmt, HOST_TO_DEVICE);
68732837fe3STobias Grosser     isl_ast_expr_free(Expr);
68813c78e4dSTobias Grosser     return;
68913c78e4dSTobias Grosser   }
69013c78e4dSTobias Grosser 
69113c78e4dSTobias Grosser   if (isPrefix(Str, "from_device")) {
69213c78e4dSTobias Grosser     createDataTransfer(UserStmt, DEVICE_TO_HOST);
69313c78e4dSTobias Grosser     isl_ast_expr_free(Expr);
69438fc0aedSTobias Grosser     return;
69538fc0aedSTobias Grosser   }
69638fc0aedSTobias Grosser 
6975260c041STobias Grosser   isl_id *Anno = isl_ast_node_get_annotation(UserStmt);
6985260c041STobias Grosser   struct ppcg_kernel_stmt *KernelStmt =
6995260c041STobias Grosser       (struct ppcg_kernel_stmt *)isl_id_get_user(Anno);
7005260c041STobias Grosser   isl_id_free(Anno);
7015260c041STobias Grosser 
7025260c041STobias Grosser   switch (KernelStmt->type) {
7035260c041STobias Grosser   case ppcg_kernel_domain:
704edb885cbSTobias Grosser     createScopStmt(Expr, KernelStmt);
7055260c041STobias Grosser     isl_ast_node_free(UserStmt);
7065260c041STobias Grosser     return;
7075260c041STobias Grosser   case ppcg_kernel_copy:
7085260c041STobias Grosser     // TODO: Create kernel copy stmt
7095260c041STobias Grosser     isl_ast_expr_free(Expr);
7105260c041STobias Grosser     isl_ast_node_free(UserStmt);
7115260c041STobias Grosser     return;
7125260c041STobias Grosser   case ppcg_kernel_sync:
7135260c041STobias Grosser     createKernelSync();
7145260c041STobias Grosser     isl_ast_expr_free(Expr);
7155260c041STobias Grosser     isl_ast_node_free(UserStmt);
7165260c041STobias Grosser     return;
7175260c041STobias Grosser   }
7185260c041STobias Grosser 
7195260c041STobias Grosser   isl_ast_expr_free(Expr);
7205260c041STobias Grosser   isl_ast_node_free(UserStmt);
7215260c041STobias Grosser   return;
7225260c041STobias Grosser }
7235260c041STobias Grosser 
724edb885cbSTobias Grosser void GPUNodeBuilder::createScopStmt(isl_ast_expr *Expr,
725edb885cbSTobias Grosser                                     ppcg_kernel_stmt *KernelStmt) {
726edb885cbSTobias Grosser   auto Stmt = (ScopStmt *)KernelStmt->u.d.stmt->stmt;
727edb885cbSTobias Grosser   isl_id_to_ast_expr *Indexes = KernelStmt->u.d.ref2expr;
728edb885cbSTobias Grosser 
729edb885cbSTobias Grosser   LoopToScevMapT LTS;
730edb885cbSTobias Grosser   LTS.insert(OutsideLoopIterations.begin(), OutsideLoopIterations.end());
731edb885cbSTobias Grosser 
732edb885cbSTobias Grosser   createSubstitutions(Expr, Stmt, LTS);
733edb885cbSTobias Grosser 
734edb885cbSTobias Grosser   if (Stmt->isBlockStmt())
735edb885cbSTobias Grosser     BlockGen.copyStmt(*Stmt, LTS, Indexes);
736edb885cbSTobias Grosser   else
737edb885cbSTobias Grosser     assert(0 && "Region statement not supported\n");
738edb885cbSTobias Grosser }
739edb885cbSTobias Grosser 
7405260c041STobias Grosser void GPUNodeBuilder::createKernelSync() {
7415260c041STobias Grosser   Module *M = Builder.GetInsertBlock()->getParent()->getParent();
7425260c041STobias Grosser   auto *Sync = Intrinsic::getDeclaration(M, Intrinsic::nvvm_barrier0);
7435260c041STobias Grosser   Builder.CreateCall(Sync, {});
7445260c041STobias Grosser }
7455260c041STobias Grosser 
746edb885cbSTobias Grosser /// Collect llvm::Values referenced from @p Node
747edb885cbSTobias Grosser ///
748edb885cbSTobias Grosser /// This function only applies to isl_ast_nodes that are user_nodes referring
749edb885cbSTobias Grosser /// to a ScopStmt. All other node types are ignore.
750edb885cbSTobias Grosser ///
751edb885cbSTobias Grosser /// @param Node The node to collect references for.
752edb885cbSTobias Grosser /// @param User A user pointer used as storage for the data that is collected.
753edb885cbSTobias Grosser ///
754edb885cbSTobias Grosser /// @returns isl_bool_true if data could be collected successfully.
755edb885cbSTobias Grosser isl_bool collectReferencesInGPUStmt(__isl_keep isl_ast_node *Node, void *User) {
756edb885cbSTobias Grosser   if (isl_ast_node_get_type(Node) != isl_ast_node_user)
757edb885cbSTobias Grosser     return isl_bool_true;
758edb885cbSTobias Grosser 
759edb885cbSTobias Grosser   isl_ast_expr *Expr = isl_ast_node_user_get_expr(Node);
760edb885cbSTobias Grosser   isl_ast_expr *StmtExpr = isl_ast_expr_get_op_arg(Expr, 0);
761edb885cbSTobias Grosser   isl_id *Id = isl_ast_expr_get_id(StmtExpr);
762edb885cbSTobias Grosser   const char *Str = isl_id_get_name(Id);
763edb885cbSTobias Grosser   isl_id_free(Id);
764edb885cbSTobias Grosser   isl_ast_expr_free(StmtExpr);
765edb885cbSTobias Grosser   isl_ast_expr_free(Expr);
766edb885cbSTobias Grosser 
767edb885cbSTobias Grosser   if (!isPrefix(Str, "Stmt"))
768edb885cbSTobias Grosser     return isl_bool_true;
769edb885cbSTobias Grosser 
770edb885cbSTobias Grosser   Id = isl_ast_node_get_annotation(Node);
771edb885cbSTobias Grosser   auto *KernelStmt = (ppcg_kernel_stmt *)isl_id_get_user(Id);
772edb885cbSTobias Grosser   auto Stmt = (ScopStmt *)KernelStmt->u.d.stmt->stmt;
773edb885cbSTobias Grosser   isl_id_free(Id);
774edb885cbSTobias Grosser 
775edb885cbSTobias Grosser   addReferencesFromStmt(Stmt, User);
776edb885cbSTobias Grosser 
777edb885cbSTobias Grosser   return isl_bool_true;
778edb885cbSTobias Grosser }
779edb885cbSTobias Grosser 
780edb885cbSTobias Grosser SetVector<Value *> GPUNodeBuilder::getReferencesInKernel(ppcg_kernel *Kernel) {
781edb885cbSTobias Grosser   SetVector<Value *> SubtreeValues;
782edb885cbSTobias Grosser   SetVector<const SCEV *> SCEVs;
783edb885cbSTobias Grosser   SetVector<const Loop *> Loops;
784edb885cbSTobias Grosser   SubtreeReferences References = {
785edb885cbSTobias Grosser       LI, SE, S, ValueMap, SubtreeValues, SCEVs, getBlockGenerator()};
786edb885cbSTobias Grosser 
787edb885cbSTobias Grosser   for (const auto &I : IDToValue)
788edb885cbSTobias Grosser     SubtreeValues.insert(I.second);
789edb885cbSTobias Grosser 
790edb885cbSTobias Grosser   isl_ast_node_foreach_descendant_top_down(
791edb885cbSTobias Grosser       Kernel->tree, collectReferencesInGPUStmt, &References);
792edb885cbSTobias Grosser 
793edb885cbSTobias Grosser   for (const SCEV *Expr : SCEVs)
794edb885cbSTobias Grosser     findValues(Expr, SE, SubtreeValues);
795edb885cbSTobias Grosser 
796edb885cbSTobias Grosser   for (auto &SAI : S.arrays())
797edb885cbSTobias Grosser     SubtreeValues.remove(SAI.second->getBasePtr());
798edb885cbSTobias Grosser 
799edb885cbSTobias Grosser   isl_space *Space = S.getParamSpace();
800edb885cbSTobias Grosser   for (long i = 0; i < isl_space_dim(Space, isl_dim_param); i++) {
801edb885cbSTobias Grosser     isl_id *Id = isl_space_get_dim_id(Space, isl_dim_param, i);
802edb885cbSTobias Grosser     assert(IDToValue.count(Id));
803edb885cbSTobias Grosser     Value *Val = IDToValue[Id];
804edb885cbSTobias Grosser     SubtreeValues.remove(Val);
805edb885cbSTobias Grosser     isl_id_free(Id);
806edb885cbSTobias Grosser   }
807edb885cbSTobias Grosser   isl_space_free(Space);
808edb885cbSTobias Grosser 
809edb885cbSTobias Grosser   for (long i = 0; i < isl_space_dim(Kernel->space, isl_dim_set); i++) {
810edb885cbSTobias Grosser     isl_id *Id = isl_space_get_dim_id(Kernel->space, isl_dim_set, i);
811edb885cbSTobias Grosser     assert(IDToValue.count(Id));
812edb885cbSTobias Grosser     Value *Val = IDToValue[Id];
813edb885cbSTobias Grosser     SubtreeValues.remove(Val);
814edb885cbSTobias Grosser     isl_id_free(Id);
815edb885cbSTobias Grosser   }
816edb885cbSTobias Grosser 
817edb885cbSTobias Grosser   return SubtreeValues;
818edb885cbSTobias Grosser }
819edb885cbSTobias Grosser 
82074dc3cb4STobias Grosser void GPUNodeBuilder::clearDominators(Function *F) {
82174dc3cb4STobias Grosser   DomTreeNode *N = DT.getNode(&F->getEntryBlock());
82274dc3cb4STobias Grosser   std::vector<BasicBlock *> Nodes;
82374dc3cb4STobias Grosser   for (po_iterator<DomTreeNode *> I = po_begin(N), E = po_end(N); I != E; ++I)
82474dc3cb4STobias Grosser     Nodes.push_back(I->getBlock());
82574dc3cb4STobias Grosser 
82674dc3cb4STobias Grosser   for (BasicBlock *BB : Nodes)
82774dc3cb4STobias Grosser     DT.eraseNode(BB);
82874dc3cb4STobias Grosser }
82974dc3cb4STobias Grosser 
83074dc3cb4STobias Grosser void GPUNodeBuilder::clearScalarEvolution(Function *F) {
83174dc3cb4STobias Grosser   for (BasicBlock &BB : *F) {
83274dc3cb4STobias Grosser     Loop *L = LI.getLoopFor(&BB);
83374dc3cb4STobias Grosser     if (L)
83474dc3cb4STobias Grosser       SE.forgetLoop(L);
83574dc3cb4STobias Grosser   }
83674dc3cb4STobias Grosser }
83774dc3cb4STobias Grosser 
83874dc3cb4STobias Grosser void GPUNodeBuilder::clearLoops(Function *F) {
83974dc3cb4STobias Grosser   for (BasicBlock &BB : *F) {
84074dc3cb4STobias Grosser     Loop *L = LI.getLoopFor(&BB);
84174dc3cb4STobias Grosser     if (L)
84274dc3cb4STobias Grosser       SE.forgetLoop(L);
84374dc3cb4STobias Grosser     LI.removeBlock(&BB);
84474dc3cb4STobias Grosser   }
84574dc3cb4STobias Grosser }
84674dc3cb4STobias Grosser 
84779a947c2STobias Grosser std::tuple<Value *, Value *> GPUNodeBuilder::getGridSizes(ppcg_kernel *Kernel) {
84879a947c2STobias Grosser   std::vector<Value *> Sizes;
84979a947c2STobias Grosser   isl_ast_build *Context = isl_ast_build_from_context(S.getContext());
85079a947c2STobias Grosser 
85179a947c2STobias Grosser   for (long i = 0; i < Kernel->n_grid; i++) {
85279a947c2STobias Grosser     isl_pw_aff *Size = isl_multi_pw_aff_get_pw_aff(Kernel->grid_size, i);
85379a947c2STobias Grosser     isl_ast_expr *GridSize = isl_ast_build_expr_from_pw_aff(Context, Size);
85479a947c2STobias Grosser     Value *Res = ExprBuilder.create(GridSize);
85579a947c2STobias Grosser     Res = Builder.CreateTrunc(Res, Builder.getInt32Ty());
85679a947c2STobias Grosser     Sizes.push_back(Res);
85779a947c2STobias Grosser   }
85879a947c2STobias Grosser   isl_ast_build_free(Context);
85979a947c2STobias Grosser 
86079a947c2STobias Grosser   for (long i = Kernel->n_grid; i < 3; i++)
86179a947c2STobias Grosser     Sizes.push_back(ConstantInt::get(Builder.getInt32Ty(), 1));
86279a947c2STobias Grosser 
86379a947c2STobias Grosser   return std::make_tuple(Sizes[0], Sizes[1]);
86479a947c2STobias Grosser }
86579a947c2STobias Grosser 
86679a947c2STobias Grosser std::tuple<Value *, Value *, Value *>
86779a947c2STobias Grosser GPUNodeBuilder::getBlockSizes(ppcg_kernel *Kernel) {
86879a947c2STobias Grosser   std::vector<Value *> Sizes;
86979a947c2STobias Grosser 
87079a947c2STobias Grosser   for (long i = 0; i < Kernel->n_block; i++) {
87179a947c2STobias Grosser     Value *Res = ConstantInt::get(Builder.getInt32Ty(), Kernel->block_dim[i]);
87279a947c2STobias Grosser     Sizes.push_back(Res);
87379a947c2STobias Grosser   }
87479a947c2STobias Grosser 
87579a947c2STobias Grosser   for (long i = Kernel->n_block; i < 3; i++)
87679a947c2STobias Grosser     Sizes.push_back(ConstantInt::get(Builder.getInt32Ty(), 1));
87779a947c2STobias Grosser 
87879a947c2STobias Grosser   return std::make_tuple(Sizes[0], Sizes[1], Sizes[2]);
87979a947c2STobias Grosser }
88079a947c2STobias Grosser 
88179a947c2STobias Grosser Value *GPUNodeBuilder::createLaunchParameters(ppcg_kernel *Kernel,
88279a947c2STobias Grosser                                               Function *F) {
883*4e18d71cSTobias Grosser   Type *ArrayTy = ArrayType::get(Builder.getInt8PtrTy(),
884*4e18d71cSTobias Grosser                                  std::distance(F->arg_begin(), F->arg_end()));
88579a947c2STobias Grosser 
88679a947c2STobias Grosser   BasicBlock *EntryBlock =
88779a947c2STobias Grosser       &Builder.GetInsertBlock()->getParent()->getEntryBlock();
88879a947c2STobias Grosser   std::string Launch = "polly_launch_" + std::to_string(Kernel->id);
88979a947c2STobias Grosser   Instruction *Parameters =
89079a947c2STobias Grosser       new AllocaInst(ArrayTy, Launch + "_params", EntryBlock->getTerminator());
89179a947c2STobias Grosser 
89279a947c2STobias Grosser   int Index = 0;
89379a947c2STobias Grosser   for (long i = 0; i < Prog->n_array; i++) {
89479a947c2STobias Grosser     if (!ppcg_kernel_requires_array_argument(Kernel, i))
89579a947c2STobias Grosser       continue;
89679a947c2STobias Grosser 
89779a947c2STobias Grosser     isl_id *Id = isl_space_get_tuple_id(Prog->array[i].space, isl_dim_set);
89879a947c2STobias Grosser     const ScopArrayInfo *SAI = ScopArrayInfo::getFromId(Id);
89979a947c2STobias Grosser 
90079a947c2STobias Grosser     Value *DevArray = DeviceAllocations[(ScopArrayInfo *)SAI];
90179a947c2STobias Grosser     DevArray = createCallGetDevicePtr(DevArray);
90279a947c2STobias Grosser     Instruction *Param = new AllocaInst(
90379a947c2STobias Grosser         Builder.getInt8PtrTy(), Launch + "_param_" + std::to_string(Index),
90479a947c2STobias Grosser         EntryBlock->getTerminator());
90579a947c2STobias Grosser     Builder.CreateStore(DevArray, Param);
90679a947c2STobias Grosser     Value *Slot = Builder.CreateGEP(Parameters,
90779a947c2STobias Grosser                                     {Builder.getInt64(0), Builder.getInt64(i)});
90879a947c2STobias Grosser     Value *ParamTyped =
90979a947c2STobias Grosser         Builder.CreatePointerCast(Param, Builder.getInt8PtrTy());
91079a947c2STobias Grosser     Builder.CreateStore(ParamTyped, Slot);
91179a947c2STobias Grosser     Index++;
91279a947c2STobias Grosser   }
91379a947c2STobias Grosser 
91479a947c2STobias Grosser   auto Location = EntryBlock->getTerminator();
91579a947c2STobias Grosser   return new BitCastInst(Parameters, Builder.getInt8PtrTy(),
91679a947c2STobias Grosser                          Launch + "_params_i8ptr", Location);
91779a947c2STobias Grosser }
91879a947c2STobias Grosser 
91932837fe3STobias Grosser void GPUNodeBuilder::createKernel(__isl_take isl_ast_node *KernelStmt) {
92032837fe3STobias Grosser   isl_id *Id = isl_ast_node_get_annotation(KernelStmt);
92132837fe3STobias Grosser   ppcg_kernel *Kernel = (ppcg_kernel *)isl_id_get_user(Id);
92232837fe3STobias Grosser   isl_id_free(Id);
92332837fe3STobias Grosser   isl_ast_node_free(KernelStmt);
92432837fe3STobias Grosser 
925edb885cbSTobias Grosser   SetVector<Value *> SubtreeValues = getReferencesInKernel(Kernel);
926edb885cbSTobias Grosser 
92732837fe3STobias Grosser   assert(Kernel->tree && "Device AST of kernel node is empty");
92832837fe3STobias Grosser 
92932837fe3STobias Grosser   Instruction &HostInsertPoint = *Builder.GetInsertPoint();
930472f9654STobias Grosser   IslExprBuilder::IDToValueTy HostIDs = IDToValue;
931edb885cbSTobias Grosser   ValueMapT HostValueMap = ValueMap;
93232837fe3STobias Grosser 
933edb885cbSTobias Grosser   SetVector<const Loop *> Loops;
934edb885cbSTobias Grosser 
935edb885cbSTobias Grosser   // Create for all loops we depend on values that contain the current loop
936edb885cbSTobias Grosser   // iteration. These values are necessary to generate code for SCEVs that
937edb885cbSTobias Grosser   // depend on such loops. As a result we need to pass them to the subfunction.
938edb885cbSTobias Grosser   for (const Loop *L : Loops) {
939edb885cbSTobias Grosser     const SCEV *OuterLIV = SE.getAddRecExpr(SE.getUnknown(Builder.getInt64(0)),
940edb885cbSTobias Grosser                                             SE.getUnknown(Builder.getInt64(1)),
941edb885cbSTobias Grosser                                             L, SCEV::FlagAnyWrap);
942edb885cbSTobias Grosser     Value *V = generateSCEV(OuterLIV);
943edb885cbSTobias Grosser     OutsideLoopIterations[L] = SE.getUnknown(V);
944edb885cbSTobias Grosser     SubtreeValues.insert(V);
945edb885cbSTobias Grosser   }
946edb885cbSTobias Grosser 
947edb885cbSTobias Grosser   createKernelFunction(Kernel, SubtreeValues);
94832837fe3STobias Grosser 
94959ab0705STobias Grosser   create(isl_ast_node_copy(Kernel->tree));
95059ab0705STobias Grosser 
95174dc3cb4STobias Grosser   Function *F = Builder.GetInsertBlock()->getParent();
95274dc3cb4STobias Grosser   clearDominators(F);
95374dc3cb4STobias Grosser   clearScalarEvolution(F);
95474dc3cb4STobias Grosser   clearLoops(F);
95574dc3cb4STobias Grosser 
95632837fe3STobias Grosser   Builder.SetInsertPoint(&HostInsertPoint);
957472f9654STobias Grosser   IDToValue = HostIDs;
95832837fe3STobias Grosser 
959edb885cbSTobias Grosser   ValueMap = HostValueMap;
960edb885cbSTobias Grosser   ScalarMap.clear();
961edb885cbSTobias Grosser   PHIOpMap.clear();
962edb885cbSTobias Grosser   EscapeMap.clear();
963edb885cbSTobias Grosser   IDToSAI.clear();
96474dc3cb4STobias Grosser   Annotator.resetAlternativeAliasBases();
96574dc3cb4STobias Grosser   for (auto &BasePtr : LocalArrays)
96674dc3cb4STobias Grosser     S.invalidateScopArrayInfo(BasePtr, ScopArrayInfo::MK_Array);
96774dc3cb4STobias Grosser   LocalArrays.clear();
968edb885cbSTobias Grosser 
96979a947c2STobias Grosser   Value *Parameters = createLaunchParameters(Kernel, F);
97079a947c2STobias Grosser 
97157793596STobias Grosser   std::string ASMString = finalizeKernelFunction();
97257793596STobias Grosser   std::string Name = "kernel_" + std::to_string(Kernel->id);
97357793596STobias Grosser   Value *KernelString = Builder.CreateGlobalStringPtr(ASMString, Name);
97457793596STobias Grosser   Value *NameString = Builder.CreateGlobalStringPtr(Name, Name + "_name");
97557793596STobias Grosser   Value *GPUKernel = createCallGetKernel(KernelString, NameString);
97679a947c2STobias Grosser 
97779a947c2STobias Grosser   Value *GridDimX, *GridDimY;
97879a947c2STobias Grosser   std::tie(GridDimX, GridDimY) = getGridSizes(Kernel);
97979a947c2STobias Grosser 
98079a947c2STobias Grosser   Value *BlockDimX, *BlockDimY, *BlockDimZ;
98179a947c2STobias Grosser   std::tie(BlockDimX, BlockDimY, BlockDimZ) = getBlockSizes(Kernel);
98279a947c2STobias Grosser 
98379a947c2STobias Grosser   createCallLaunchKernel(GPUKernel, GridDimX, GridDimY, BlockDimX, BlockDimY,
98479a947c2STobias Grosser                          BlockDimZ, Parameters);
98557793596STobias Grosser   createCallFreeKernel(GPUKernel);
98632837fe3STobias Grosser }
98732837fe3STobias Grosser 
98832837fe3STobias Grosser /// Compute the DataLayout string for the NVPTX backend.
98932837fe3STobias Grosser ///
99032837fe3STobias Grosser /// @param is64Bit Are we looking for a 64 bit architecture?
99132837fe3STobias Grosser static std::string computeNVPTXDataLayout(bool is64Bit) {
99232837fe3STobias Grosser   std::string Ret = "e";
99332837fe3STobias Grosser 
99432837fe3STobias Grosser   if (!is64Bit)
99532837fe3STobias Grosser     Ret += "-p:32:32";
99632837fe3STobias Grosser 
99732837fe3STobias Grosser   Ret += "-i64:64-v16:16-v32:32-n16:32:64";
99832837fe3STobias Grosser 
99932837fe3STobias Grosser   return Ret;
100032837fe3STobias Grosser }
100132837fe3STobias Grosser 
1002edb885cbSTobias Grosser Function *
1003edb885cbSTobias Grosser GPUNodeBuilder::createKernelFunctionDecl(ppcg_kernel *Kernel,
1004edb885cbSTobias Grosser                                          SetVector<Value *> &SubtreeValues) {
100532837fe3STobias Grosser   std::vector<Type *> Args;
100632837fe3STobias Grosser   std::string Identifier = "kernel_" + std::to_string(Kernel->id);
100732837fe3STobias Grosser 
100832837fe3STobias Grosser   for (long i = 0; i < Prog->n_array; i++) {
100932837fe3STobias Grosser     if (!ppcg_kernel_requires_array_argument(Kernel, i))
101032837fe3STobias Grosser       continue;
101132837fe3STobias Grosser 
101232837fe3STobias Grosser     Args.push_back(Builder.getInt8PtrTy());
101332837fe3STobias Grosser   }
101432837fe3STobias Grosser 
1015f6044bd0STobias Grosser   int NumHostIters = isl_space_dim(Kernel->space, isl_dim_set);
1016f6044bd0STobias Grosser 
1017f6044bd0STobias Grosser   for (long i = 0; i < NumHostIters; i++)
1018f6044bd0STobias Grosser     Args.push_back(Builder.getInt64Ty());
1019f6044bd0STobias Grosser 
1020c84a1995STobias Grosser   int NumVars = isl_space_dim(Kernel->space, isl_dim_param);
1021c84a1995STobias Grosser 
1022c84a1995STobias Grosser   for (long i = 0; i < NumVars; i++)
1023c84a1995STobias Grosser     Args.push_back(Builder.getInt64Ty());
1024c84a1995STobias Grosser 
1025edb885cbSTobias Grosser   for (auto *V : SubtreeValues)
1026edb885cbSTobias Grosser     Args.push_back(V->getType());
1027edb885cbSTobias Grosser 
102832837fe3STobias Grosser   auto *FT = FunctionType::get(Builder.getVoidTy(), Args, false);
102932837fe3STobias Grosser   auto *FN = Function::Create(FT, Function::ExternalLinkage, Identifier,
103032837fe3STobias Grosser                               GPUModule.get());
103132837fe3STobias Grosser   FN->setCallingConv(CallingConv::PTX_Kernel);
103232837fe3STobias Grosser 
103332837fe3STobias Grosser   auto Arg = FN->arg_begin();
103432837fe3STobias Grosser   for (long i = 0; i < Kernel->n_array; i++) {
103532837fe3STobias Grosser     if (!ppcg_kernel_requires_array_argument(Kernel, i))
103632837fe3STobias Grosser       continue;
103732837fe3STobias Grosser 
1038edb885cbSTobias Grosser     Arg->setName(Kernel->array[i].array->name);
1039edb885cbSTobias Grosser 
1040edb885cbSTobias Grosser     isl_id *Id = isl_space_get_tuple_id(Prog->array[i].space, isl_dim_set);
1041edb885cbSTobias Grosser     const ScopArrayInfo *SAI = ScopArrayInfo::getFromId(isl_id_copy(Id));
1042edb885cbSTobias Grosser     Type *EleTy = SAI->getElementType();
1043edb885cbSTobias Grosser     Value *Val = &*Arg;
1044edb885cbSTobias Grosser     SmallVector<const SCEV *, 4> Sizes;
1045edb885cbSTobias Grosser     isl_ast_build *Build =
1046edb885cbSTobias Grosser         isl_ast_build_from_context(isl_set_copy(Prog->context));
1047edb885cbSTobias Grosser     for (long j = 1; j < Kernel->array[i].array->n_index; j++) {
1048edb885cbSTobias Grosser       isl_ast_expr *DimSize = isl_ast_build_expr_from_pw_aff(
1049edb885cbSTobias Grosser           Build, isl_pw_aff_copy(Kernel->array[i].array->bound[j]));
1050edb885cbSTobias Grosser       auto V = ExprBuilder.create(DimSize);
1051edb885cbSTobias Grosser       Sizes.push_back(SE.getSCEV(V));
1052edb885cbSTobias Grosser     }
1053edb885cbSTobias Grosser     const ScopArrayInfo *SAIRep =
1054edb885cbSTobias Grosser         S.getOrCreateScopArrayInfo(Val, EleTy, Sizes, ScopArrayInfo::MK_Array);
105574dc3cb4STobias Grosser     LocalArrays.push_back(Val);
1056edb885cbSTobias Grosser 
1057edb885cbSTobias Grosser     isl_ast_build_free(Build);
1058edb885cbSTobias Grosser     isl_id_free(Id);
1059edb885cbSTobias Grosser     IDToSAI[Id] = SAIRep;
106032837fe3STobias Grosser     Arg++;
106132837fe3STobias Grosser   }
106232837fe3STobias Grosser 
1063f6044bd0STobias Grosser   for (long i = 0; i < NumHostIters; i++) {
1064f6044bd0STobias Grosser     isl_id *Id = isl_space_get_dim_id(Kernel->space, isl_dim_set, i);
1065f6044bd0STobias Grosser     Arg->setName(isl_id_get_name(Id));
1066f6044bd0STobias Grosser     IDToValue[Id] = &*Arg;
1067f6044bd0STobias Grosser     KernelIDs.insert(std::unique_ptr<isl_id, IslIdDeleter>(Id));
1068f6044bd0STobias Grosser     Arg++;
1069f6044bd0STobias Grosser   }
1070f6044bd0STobias Grosser 
1071c84a1995STobias Grosser   for (long i = 0; i < NumVars; i++) {
1072c84a1995STobias Grosser     isl_id *Id = isl_space_get_dim_id(Kernel->space, isl_dim_param, i);
1073c84a1995STobias Grosser     Arg->setName(isl_id_get_name(Id));
1074c84a1995STobias Grosser     IDToValue[Id] = &*Arg;
1075c84a1995STobias Grosser     KernelIDs.insert(std::unique_ptr<isl_id, IslIdDeleter>(Id));
1076c84a1995STobias Grosser     Arg++;
1077c84a1995STobias Grosser   }
1078c84a1995STobias Grosser 
1079edb885cbSTobias Grosser   for (auto *V : SubtreeValues) {
1080edb885cbSTobias Grosser     Arg->setName(V->getName());
1081edb885cbSTobias Grosser     ValueMap[V] = &*Arg;
1082edb885cbSTobias Grosser     Arg++;
1083edb885cbSTobias Grosser   }
1084edb885cbSTobias Grosser 
108532837fe3STobias Grosser   return FN;
108632837fe3STobias Grosser }
108732837fe3STobias Grosser 
1088472f9654STobias Grosser void GPUNodeBuilder::insertKernelIntrinsics(ppcg_kernel *Kernel) {
1089472f9654STobias Grosser   Intrinsic::ID IntrinsicsBID[] = {Intrinsic::nvvm_read_ptx_sreg_ctaid_x,
1090472f9654STobias Grosser                                    Intrinsic::nvvm_read_ptx_sreg_ctaid_y};
1091472f9654STobias Grosser 
1092472f9654STobias Grosser   Intrinsic::ID IntrinsicsTID[] = {Intrinsic::nvvm_read_ptx_sreg_tid_x,
1093472f9654STobias Grosser                                    Intrinsic::nvvm_read_ptx_sreg_tid_y,
1094472f9654STobias Grosser                                    Intrinsic::nvvm_read_ptx_sreg_tid_z};
1095472f9654STobias Grosser 
1096472f9654STobias Grosser   auto addId = [this](__isl_take isl_id *Id, Intrinsic::ID Intr) mutable {
1097472f9654STobias Grosser     std::string Name = isl_id_get_name(Id);
1098472f9654STobias Grosser     Module *M = Builder.GetInsertBlock()->getParent()->getParent();
1099472f9654STobias Grosser     Function *IntrinsicFn = Intrinsic::getDeclaration(M, Intr);
1100472f9654STobias Grosser     Value *Val = Builder.CreateCall(IntrinsicFn, {});
1101472f9654STobias Grosser     Val = Builder.CreateIntCast(Val, Builder.getInt64Ty(), false, Name);
1102472f9654STobias Grosser     IDToValue[Id] = Val;
1103472f9654STobias Grosser     KernelIDs.insert(std::unique_ptr<isl_id, IslIdDeleter>(Id));
1104472f9654STobias Grosser   };
1105472f9654STobias Grosser 
1106472f9654STobias Grosser   for (int i = 0; i < Kernel->n_grid; ++i) {
1107472f9654STobias Grosser     isl_id *Id = isl_id_list_get_id(Kernel->block_ids, i);
1108472f9654STobias Grosser     addId(Id, IntrinsicsBID[i]);
1109472f9654STobias Grosser   }
1110472f9654STobias Grosser 
1111472f9654STobias Grosser   for (int i = 0; i < Kernel->n_block; ++i) {
1112472f9654STobias Grosser     isl_id *Id = isl_id_list_get_id(Kernel->thread_ids, i);
1113472f9654STobias Grosser     addId(Id, IntrinsicsTID[i]);
1114472f9654STobias Grosser   }
1115472f9654STobias Grosser }
1116472f9654STobias Grosser 
1117edb885cbSTobias Grosser void GPUNodeBuilder::createKernelFunction(ppcg_kernel *Kernel,
1118edb885cbSTobias Grosser                                           SetVector<Value *> &SubtreeValues) {
111932837fe3STobias Grosser 
112032837fe3STobias Grosser   std::string Identifier = "kernel_" + std::to_string(Kernel->id);
112132837fe3STobias Grosser   GPUModule.reset(new Module(Identifier, Builder.getContext()));
112232837fe3STobias Grosser   GPUModule->setTargetTriple(Triple::normalize("nvptx64-nvidia-cuda"));
112332837fe3STobias Grosser   GPUModule->setDataLayout(computeNVPTXDataLayout(true /* is64Bit */));
112432837fe3STobias Grosser 
1125edb885cbSTobias Grosser   Function *FN = createKernelFunctionDecl(Kernel, SubtreeValues);
112632837fe3STobias Grosser 
112759ab0705STobias Grosser   BasicBlock *PrevBlock = Builder.GetInsertBlock();
112832837fe3STobias Grosser   auto EntryBlock = BasicBlock::Create(Builder.getContext(), "entry", FN);
112932837fe3STobias Grosser 
113059ab0705STobias Grosser   DominatorTree &DT = P->getAnalysis<DominatorTreeWrapperPass>().getDomTree();
113159ab0705STobias Grosser   DT.addNewBlock(EntryBlock, PrevBlock);
113259ab0705STobias Grosser 
113332837fe3STobias Grosser   Builder.SetInsertPoint(EntryBlock);
113432837fe3STobias Grosser   Builder.CreateRetVoid();
113532837fe3STobias Grosser   Builder.SetInsertPoint(EntryBlock, EntryBlock->begin());
1136472f9654STobias Grosser 
1137472f9654STobias Grosser   insertKernelIntrinsics(Kernel);
113832837fe3STobias Grosser }
113932837fe3STobias Grosser 
114074dc3cb4STobias Grosser std::string GPUNodeBuilder::createKernelASM() {
114174dc3cb4STobias Grosser   llvm::Triple GPUTriple(Triple::normalize("nvptx64-nvidia-cuda"));
114274dc3cb4STobias Grosser   std::string ErrMsg;
114374dc3cb4STobias Grosser   auto GPUTarget = TargetRegistry::lookupTarget(GPUTriple.getTriple(), ErrMsg);
114474dc3cb4STobias Grosser 
114574dc3cb4STobias Grosser   if (!GPUTarget) {
114674dc3cb4STobias Grosser     errs() << ErrMsg << "\n";
114774dc3cb4STobias Grosser     return "";
114874dc3cb4STobias Grosser   }
114974dc3cb4STobias Grosser 
115074dc3cb4STobias Grosser   TargetOptions Options;
115174dc3cb4STobias Grosser   Options.UnsafeFPMath = FastMath;
115274dc3cb4STobias Grosser   std::unique_ptr<TargetMachine> TargetM(
115374dc3cb4STobias Grosser       GPUTarget->createTargetMachine(GPUTriple.getTriple(), CudaVersion, "",
115474dc3cb4STobias Grosser                                      Options, Optional<Reloc::Model>()));
115574dc3cb4STobias Grosser 
115674dc3cb4STobias Grosser   SmallString<0> ASMString;
115774dc3cb4STobias Grosser   raw_svector_ostream ASMStream(ASMString);
115874dc3cb4STobias Grosser   llvm::legacy::PassManager PM;
115974dc3cb4STobias Grosser 
116074dc3cb4STobias Grosser   PM.add(createTargetTransformInfoWrapperPass(TargetM->getTargetIRAnalysis()));
116174dc3cb4STobias Grosser 
116274dc3cb4STobias Grosser   if (TargetM->addPassesToEmitFile(
116374dc3cb4STobias Grosser           PM, ASMStream, TargetMachine::CGFT_AssemblyFile, true /* verify */)) {
116474dc3cb4STobias Grosser     errs() << "The target does not support generation of this file type!\n";
116574dc3cb4STobias Grosser     return "";
116674dc3cb4STobias Grosser   }
116774dc3cb4STobias Grosser 
116874dc3cb4STobias Grosser   PM.run(*GPUModule);
116974dc3cb4STobias Grosser 
117074dc3cb4STobias Grosser   return ASMStream.str();
117174dc3cb4STobias Grosser }
117274dc3cb4STobias Grosser 
117357793596STobias Grosser std::string GPUNodeBuilder::finalizeKernelFunction() {
1174e1a98343STobias Grosser   // Verify module.
1175e1a98343STobias Grosser   llvm::legacy::PassManager Passes;
1176e1a98343STobias Grosser   Passes.add(createVerifierPass());
1177e1a98343STobias Grosser   Passes.run(*GPUModule);
117832837fe3STobias Grosser 
117932837fe3STobias Grosser   if (DumpKernelIR)
118032837fe3STobias Grosser     outs() << *GPUModule << "\n";
118132837fe3STobias Grosser 
11829a18d559STobias Grosser   // Optimize module.
11839a18d559STobias Grosser   llvm::legacy::PassManager OptPasses;
11849a18d559STobias Grosser   PassManagerBuilder PassBuilder;
11859a18d559STobias Grosser   PassBuilder.OptLevel = 3;
11869a18d559STobias Grosser   PassBuilder.SizeLevel = 0;
11879a18d559STobias Grosser   PassBuilder.populateModulePassManager(OptPasses);
11889a18d559STobias Grosser   OptPasses.run(*GPUModule);
11899a18d559STobias Grosser 
119074dc3cb4STobias Grosser   std::string Assembly = createKernelASM();
119174dc3cb4STobias Grosser 
119274dc3cb4STobias Grosser   if (DumpKernelASM)
119374dc3cb4STobias Grosser     outs() << Assembly << "\n";
119474dc3cb4STobias Grosser 
119532837fe3STobias Grosser   GPUModule.release();
1196472f9654STobias Grosser   KernelIDs.clear();
119757793596STobias Grosser 
119857793596STobias Grosser   return Assembly;
119932837fe3STobias Grosser }
120032837fe3STobias Grosser 
12019dfe4e7cSTobias Grosser namespace {
12029dfe4e7cSTobias Grosser class PPCGCodeGeneration : public ScopPass {
12039dfe4e7cSTobias Grosser public:
12049dfe4e7cSTobias Grosser   static char ID;
12059dfe4e7cSTobias Grosser 
1206e938517eSTobias Grosser   /// The scop that is currently processed.
1207e938517eSTobias Grosser   Scop *S;
1208e938517eSTobias Grosser 
120938fc0aedSTobias Grosser   LoopInfo *LI;
121038fc0aedSTobias Grosser   DominatorTree *DT;
121138fc0aedSTobias Grosser   ScalarEvolution *SE;
121238fc0aedSTobias Grosser   const DataLayout *DL;
121338fc0aedSTobias Grosser   RegionInfo *RI;
121438fc0aedSTobias Grosser 
12159dfe4e7cSTobias Grosser   PPCGCodeGeneration() : ScopPass(ID) {}
12169dfe4e7cSTobias Grosser 
1217e938517eSTobias Grosser   /// Construct compilation options for PPCG.
1218e938517eSTobias Grosser   ///
1219e938517eSTobias Grosser   /// @returns The compilation options.
1220e938517eSTobias Grosser   ppcg_options *createPPCGOptions() {
1221e938517eSTobias Grosser     auto DebugOptions =
1222e938517eSTobias Grosser         (ppcg_debug_options *)malloc(sizeof(ppcg_debug_options));
1223e938517eSTobias Grosser     auto Options = (ppcg_options *)malloc(sizeof(ppcg_options));
1224e938517eSTobias Grosser 
1225e938517eSTobias Grosser     DebugOptions->dump_schedule_constraints = false;
1226e938517eSTobias Grosser     DebugOptions->dump_schedule = false;
1227e938517eSTobias Grosser     DebugOptions->dump_final_schedule = false;
1228e938517eSTobias Grosser     DebugOptions->dump_sizes = false;
1229e938517eSTobias Grosser 
1230e938517eSTobias Grosser     Options->debug = DebugOptions;
1231e938517eSTobias Grosser 
1232e938517eSTobias Grosser     Options->reschedule = true;
1233e938517eSTobias Grosser     Options->scale_tile_loops = false;
1234e938517eSTobias Grosser     Options->wrap = false;
1235e938517eSTobias Grosser 
1236e938517eSTobias Grosser     Options->non_negative_parameters = false;
1237e938517eSTobias Grosser     Options->ctx = nullptr;
1238e938517eSTobias Grosser     Options->sizes = nullptr;
1239e938517eSTobias Grosser 
12404eaedde5STobias Grosser     Options->tile_size = 32;
12414eaedde5STobias Grosser 
1242e938517eSTobias Grosser     Options->use_private_memory = false;
1243e938517eSTobias Grosser     Options->use_shared_memory = false;
1244e938517eSTobias Grosser     Options->max_shared_memory = 0;
1245e938517eSTobias Grosser 
1246e938517eSTobias Grosser     Options->target = PPCG_TARGET_CUDA;
1247e938517eSTobias Grosser     Options->openmp = false;
1248e938517eSTobias Grosser     Options->linearize_device_arrays = true;
1249e938517eSTobias Grosser     Options->live_range_reordering = false;
1250e938517eSTobias Grosser 
1251e938517eSTobias Grosser     Options->opencl_compiler_options = nullptr;
1252e938517eSTobias Grosser     Options->opencl_use_gpu = false;
1253e938517eSTobias Grosser     Options->opencl_n_include_file = 0;
1254e938517eSTobias Grosser     Options->opencl_include_files = nullptr;
1255e938517eSTobias Grosser     Options->opencl_print_kernel_types = false;
1256e938517eSTobias Grosser     Options->opencl_embed_kernel_code = false;
1257e938517eSTobias Grosser 
1258e938517eSTobias Grosser     Options->save_schedule_file = nullptr;
1259e938517eSTobias Grosser     Options->load_schedule_file = nullptr;
1260e938517eSTobias Grosser 
1261e938517eSTobias Grosser     return Options;
1262e938517eSTobias Grosser   }
1263e938517eSTobias Grosser 
1264f384594dSTobias Grosser   /// Get a tagged access relation containing all accesses of type @p AccessTy.
1265f384594dSTobias Grosser   ///
1266f384594dSTobias Grosser   /// Instead of a normal access of the form:
1267f384594dSTobias Grosser   ///
1268f384594dSTobias Grosser   ///   Stmt[i,j,k] -> Array[f_0(i,j,k), f_1(i,j,k)]
1269f384594dSTobias Grosser   ///
1270f384594dSTobias Grosser   /// a tagged access has the form
1271f384594dSTobias Grosser   ///
1272f384594dSTobias Grosser   ///   [Stmt[i,j,k] -> id[]] -> Array[f_0(i,j,k), f_1(i,j,k)]
1273f384594dSTobias Grosser   ///
1274f384594dSTobias Grosser   /// where 'id' is an additional space that references the memory access that
1275f384594dSTobias Grosser   /// triggered the access.
1276f384594dSTobias Grosser   ///
1277f384594dSTobias Grosser   /// @param AccessTy The type of the memory accesses to collect.
1278f384594dSTobias Grosser   ///
1279f384594dSTobias Grosser   /// @return The relation describing all tagged memory accesses.
1280f384594dSTobias Grosser   isl_union_map *getTaggedAccesses(enum MemoryAccess::AccessType AccessTy) {
1281f384594dSTobias Grosser     isl_union_map *Accesses = isl_union_map_empty(S->getParamSpace());
1282f384594dSTobias Grosser 
1283f384594dSTobias Grosser     for (auto &Stmt : *S)
1284f384594dSTobias Grosser       for (auto &Acc : Stmt)
1285f384594dSTobias Grosser         if (Acc->getType() == AccessTy) {
1286f384594dSTobias Grosser           isl_map *Relation = Acc->getAccessRelation();
1287f384594dSTobias Grosser           Relation = isl_map_intersect_domain(Relation, Stmt.getDomain());
1288f384594dSTobias Grosser 
1289f384594dSTobias Grosser           isl_space *Space = isl_map_get_space(Relation);
1290f384594dSTobias Grosser           Space = isl_space_range(Space);
1291f384594dSTobias Grosser           Space = isl_space_from_range(Space);
12926293ba69STobias Grosser           Space = isl_space_set_tuple_id(Space, isl_dim_in, Acc->getId());
1293f384594dSTobias Grosser           isl_map *Universe = isl_map_universe(Space);
1294f384594dSTobias Grosser           Relation = isl_map_domain_product(Relation, Universe);
1295f384594dSTobias Grosser           Accesses = isl_union_map_add_map(Accesses, Relation);
1296f384594dSTobias Grosser         }
1297f384594dSTobias Grosser 
1298f384594dSTobias Grosser     return Accesses;
1299f384594dSTobias Grosser   }
1300f384594dSTobias Grosser 
1301f384594dSTobias Grosser   /// Get the set of all read accesses, tagged with the access id.
1302f384594dSTobias Grosser   ///
1303f384594dSTobias Grosser   /// @see getTaggedAccesses
1304f384594dSTobias Grosser   isl_union_map *getTaggedReads() {
1305f384594dSTobias Grosser     return getTaggedAccesses(MemoryAccess::READ);
1306f384594dSTobias Grosser   }
1307f384594dSTobias Grosser 
1308f384594dSTobias Grosser   /// Get the set of all may (and must) accesses, tagged with the access id.
1309f384594dSTobias Grosser   ///
1310f384594dSTobias Grosser   /// @see getTaggedAccesses
1311f384594dSTobias Grosser   isl_union_map *getTaggedMayWrites() {
1312f384594dSTobias Grosser     return isl_union_map_union(getTaggedAccesses(MemoryAccess::MAY_WRITE),
1313f384594dSTobias Grosser                                getTaggedAccesses(MemoryAccess::MUST_WRITE));
1314f384594dSTobias Grosser   }
1315f384594dSTobias Grosser 
1316f384594dSTobias Grosser   /// Get the set of all must accesses, tagged with the access id.
1317f384594dSTobias Grosser   ///
1318f384594dSTobias Grosser   /// @see getTaggedAccesses
1319f384594dSTobias Grosser   isl_union_map *getTaggedMustWrites() {
1320f384594dSTobias Grosser     return getTaggedAccesses(MemoryAccess::MUST_WRITE);
1321f384594dSTobias Grosser   }
1322f384594dSTobias Grosser 
1323aef5196fSTobias Grosser   /// Collect parameter and array names as isl_ids.
1324aef5196fSTobias Grosser   ///
1325aef5196fSTobias Grosser   /// To reason about the different parameters and arrays used, ppcg requires
1326aef5196fSTobias Grosser   /// a list of all isl_ids in use. As PPCG traditionally performs
1327aef5196fSTobias Grosser   /// source-to-source compilation each of these isl_ids is mapped to the
1328aef5196fSTobias Grosser   /// expression that represents it. As we do not have a corresponding
1329aef5196fSTobias Grosser   /// expression in Polly, we just map each id to a 'zero' expression to match
1330aef5196fSTobias Grosser   /// the data format that ppcg expects.
1331aef5196fSTobias Grosser   ///
1332aef5196fSTobias Grosser   /// @returns Retun a map from collected ids to 'zero' ast expressions.
1333aef5196fSTobias Grosser   __isl_give isl_id_to_ast_expr *getNames() {
1334aef5196fSTobias Grosser     auto *Names = isl_id_to_ast_expr_alloc(
1335bd81a7eeSTobias Grosser         S->getIslCtx(),
1336bd81a7eeSTobias Grosser         S->getNumParams() + std::distance(S->array_begin(), S->array_end()));
1337aef5196fSTobias Grosser     auto *Zero = isl_ast_expr_from_val(isl_val_zero(S->getIslCtx()));
1338aef5196fSTobias Grosser     auto *Space = S->getParamSpace();
1339aef5196fSTobias Grosser 
1340aef5196fSTobias Grosser     for (int I = 0, E = S->getNumParams(); I < E; ++I) {
1341aef5196fSTobias Grosser       isl_id *Id = isl_space_get_dim_id(Space, isl_dim_param, I);
1342aef5196fSTobias Grosser       Names = isl_id_to_ast_expr_set(Names, Id, isl_ast_expr_copy(Zero));
1343aef5196fSTobias Grosser     }
1344aef5196fSTobias Grosser 
1345aef5196fSTobias Grosser     for (auto &Array : S->arrays()) {
1346aef5196fSTobias Grosser       auto Id = Array.second->getBasePtrId();
1347aef5196fSTobias Grosser       Names = isl_id_to_ast_expr_set(Names, Id, isl_ast_expr_copy(Zero));
1348aef5196fSTobias Grosser     }
1349aef5196fSTobias Grosser 
1350aef5196fSTobias Grosser     isl_space_free(Space);
1351aef5196fSTobias Grosser     isl_ast_expr_free(Zero);
1352aef5196fSTobias Grosser 
1353aef5196fSTobias Grosser     return Names;
1354aef5196fSTobias Grosser   }
1355aef5196fSTobias Grosser 
1356e938517eSTobias Grosser   /// Create a new PPCG scop from the current scop.
1357e938517eSTobias Grosser   ///
1358f384594dSTobias Grosser   /// The PPCG scop is initialized with data from the current polly::Scop. From
1359f384594dSTobias Grosser   /// this initial data, the data-dependences in the PPCG scop are initialized.
1360f384594dSTobias Grosser   /// We do not use Polly's dependence analysis for now, to ensure we match
1361f384594dSTobias Grosser   /// the PPCG default behaviour more closely.
1362e938517eSTobias Grosser   ///
1363e938517eSTobias Grosser   /// @returns A new ppcg scop.
1364e938517eSTobias Grosser   ppcg_scop *createPPCGScop() {
1365e938517eSTobias Grosser     auto PPCGScop = (ppcg_scop *)malloc(sizeof(ppcg_scop));
1366e938517eSTobias Grosser 
1367e938517eSTobias Grosser     PPCGScop->options = createPPCGOptions();
1368e938517eSTobias Grosser 
1369e938517eSTobias Grosser     PPCGScop->start = 0;
1370e938517eSTobias Grosser     PPCGScop->end = 0;
1371e938517eSTobias Grosser 
1372f384594dSTobias Grosser     PPCGScop->context = S->getContext();
1373f384594dSTobias Grosser     PPCGScop->domain = S->getDomains();
1374e938517eSTobias Grosser     PPCGScop->call = nullptr;
1375f384594dSTobias Grosser     PPCGScop->tagged_reads = getTaggedReads();
1376f384594dSTobias Grosser     PPCGScop->reads = S->getReads();
1377e938517eSTobias Grosser     PPCGScop->live_in = nullptr;
1378f384594dSTobias Grosser     PPCGScop->tagged_may_writes = getTaggedMayWrites();
1379f384594dSTobias Grosser     PPCGScop->may_writes = S->getWrites();
1380f384594dSTobias Grosser     PPCGScop->tagged_must_writes = getTaggedMustWrites();
1381f384594dSTobias Grosser     PPCGScop->must_writes = S->getMustWrites();
1382e938517eSTobias Grosser     PPCGScop->live_out = nullptr;
1383f384594dSTobias Grosser     PPCGScop->tagged_must_kills = isl_union_map_empty(S->getParamSpace());
1384e938517eSTobias Grosser     PPCGScop->tagger = nullptr;
1385e938517eSTobias Grosser 
1386e938517eSTobias Grosser     PPCGScop->independence = nullptr;
1387e938517eSTobias Grosser     PPCGScop->dep_flow = nullptr;
1388e938517eSTobias Grosser     PPCGScop->tagged_dep_flow = nullptr;
1389e938517eSTobias Grosser     PPCGScop->dep_false = nullptr;
1390e938517eSTobias Grosser     PPCGScop->dep_forced = nullptr;
1391e938517eSTobias Grosser     PPCGScop->dep_order = nullptr;
1392e938517eSTobias Grosser     PPCGScop->tagged_dep_order = nullptr;
1393e938517eSTobias Grosser 
1394f384594dSTobias Grosser     PPCGScop->schedule = S->getScheduleTree();
1395aef5196fSTobias Grosser     PPCGScop->names = getNames();
1396e938517eSTobias Grosser 
1397e938517eSTobias Grosser     PPCGScop->pet = nullptr;
1398e938517eSTobias Grosser 
1399f384594dSTobias Grosser     compute_tagger(PPCGScop);
1400f384594dSTobias Grosser     compute_dependences(PPCGScop);
1401f384594dSTobias Grosser 
1402e938517eSTobias Grosser     return PPCGScop;
1403e938517eSTobias Grosser   }
1404e938517eSTobias Grosser 
140560f63b49STobias Grosser   /// Collect the array acesses in a statement.
140660f63b49STobias Grosser   ///
140760f63b49STobias Grosser   /// @param Stmt The statement for which to collect the accesses.
140860f63b49STobias Grosser   ///
140960f63b49STobias Grosser   /// @returns A list of array accesses.
141060f63b49STobias Grosser   gpu_stmt_access *getStmtAccesses(ScopStmt &Stmt) {
141160f63b49STobias Grosser     gpu_stmt_access *Accesses = nullptr;
141260f63b49STobias Grosser 
141360f63b49STobias Grosser     for (MemoryAccess *Acc : Stmt) {
141460f63b49STobias Grosser       auto Access = isl_alloc_type(S->getIslCtx(), struct gpu_stmt_access);
141560f63b49STobias Grosser       Access->read = Acc->isRead();
141660f63b49STobias Grosser       Access->write = Acc->isWrite();
141760f63b49STobias Grosser       Access->access = Acc->getAccessRelation();
141860f63b49STobias Grosser       isl_space *Space = isl_map_get_space(Access->access);
141960f63b49STobias Grosser       Space = isl_space_range(Space);
142060f63b49STobias Grosser       Space = isl_space_from_range(Space);
14216293ba69STobias Grosser       Space = isl_space_set_tuple_id(Space, isl_dim_in, Acc->getId());
142260f63b49STobias Grosser       isl_map *Universe = isl_map_universe(Space);
142360f63b49STobias Grosser       Access->tagged_access =
142460f63b49STobias Grosser           isl_map_domain_product(Acc->getAccessRelation(), Universe);
142560f63b49STobias Grosser       Access->exact_write = Acc->isWrite();
142660f63b49STobias Grosser       Access->ref_id = Acc->getId();
142760f63b49STobias Grosser       Access->next = Accesses;
142860f63b49STobias Grosser       Accesses = Access;
142960f63b49STobias Grosser     }
143060f63b49STobias Grosser 
143160f63b49STobias Grosser     return Accesses;
143260f63b49STobias Grosser   }
143360f63b49STobias Grosser 
143469b46751STobias Grosser   /// Collect the list of GPU statements.
143569b46751STobias Grosser   ///
143669b46751STobias Grosser   /// Each statement has an id, a pointer to the underlying data structure,
143769b46751STobias Grosser   /// as well as a list with all memory accesses.
143869b46751STobias Grosser   ///
143969b46751STobias Grosser   /// TODO: Initialize the list of memory accesses.
144069b46751STobias Grosser   ///
144169b46751STobias Grosser   /// @returns A linked-list of statements.
144269b46751STobias Grosser   gpu_stmt *getStatements() {
144369b46751STobias Grosser     gpu_stmt *Stmts = isl_calloc_array(S->getIslCtx(), struct gpu_stmt,
144469b46751STobias Grosser                                        std::distance(S->begin(), S->end()));
144569b46751STobias Grosser 
144669b46751STobias Grosser     int i = 0;
144769b46751STobias Grosser     for (auto &Stmt : *S) {
144869b46751STobias Grosser       gpu_stmt *GPUStmt = &Stmts[i];
144969b46751STobias Grosser 
145069b46751STobias Grosser       GPUStmt->id = Stmt.getDomainId();
145169b46751STobias Grosser 
145269b46751STobias Grosser       // We use the pet stmt pointer to keep track of the Polly statements.
145369b46751STobias Grosser       GPUStmt->stmt = (pet_stmt *)&Stmt;
145460f63b49STobias Grosser       GPUStmt->accesses = getStmtAccesses(Stmt);
145569b46751STobias Grosser       i++;
145669b46751STobias Grosser     }
145769b46751STobias Grosser 
145869b46751STobias Grosser     return Stmts;
145969b46751STobias Grosser   }
146069b46751STobias Grosser 
146160f63b49STobias Grosser   /// Derive the extent of an array.
146260f63b49STobias Grosser   ///
146360f63b49STobias Grosser   /// The extent of an array is defined by the set of memory locations for
146460f63b49STobias Grosser   /// which a memory access in the iteration domain exists.
146560f63b49STobias Grosser   ///
146660f63b49STobias Grosser   /// @param Array The array to derive the extent for.
146760f63b49STobias Grosser   ///
146860f63b49STobias Grosser   /// @returns An isl_set describing the extent of the array.
146960f63b49STobias Grosser   __isl_give isl_set *getExtent(ScopArrayInfo *Array) {
147060f63b49STobias Grosser     isl_union_map *Accesses = S->getAccesses();
147160f63b49STobias Grosser     Accesses = isl_union_map_intersect_domain(Accesses, S->getDomains());
147260f63b49STobias Grosser     isl_union_set *AccessUSet = isl_union_map_range(Accesses);
147360f63b49STobias Grosser     isl_set *AccessSet =
147460f63b49STobias Grosser         isl_union_set_extract_set(AccessUSet, Array->getSpace());
147560f63b49STobias Grosser     isl_union_set_free(AccessUSet);
147660f63b49STobias Grosser 
147760f63b49STobias Grosser     return AccessSet;
147860f63b49STobias Grosser   }
147960f63b49STobias Grosser 
148060f63b49STobias Grosser   /// Derive the bounds of an array.
148160f63b49STobias Grosser   ///
148260f63b49STobias Grosser   /// For the first dimension we derive the bound of the array from the extent
148360f63b49STobias Grosser   /// of this dimension. For inner dimensions we obtain their size directly from
148460f63b49STobias Grosser   /// ScopArrayInfo.
148560f63b49STobias Grosser   ///
148660f63b49STobias Grosser   /// @param PPCGArray The array to compute bounds for.
148760f63b49STobias Grosser   /// @param Array The polly array from which to take the information.
148860f63b49STobias Grosser   void setArrayBounds(gpu_array_info &PPCGArray, ScopArrayInfo *Array) {
148960f63b49STobias Grosser     if (PPCGArray.n_index > 0) {
149060f63b49STobias Grosser       isl_set *Dom = isl_set_copy(PPCGArray.extent);
149160f63b49STobias Grosser       Dom = isl_set_project_out(Dom, isl_dim_set, 1, PPCGArray.n_index - 1);
149260f63b49STobias Grosser       isl_pw_aff *Bound = isl_set_dim_max(isl_set_copy(Dom), 0);
149360f63b49STobias Grosser       isl_set_free(Dom);
149460f63b49STobias Grosser       Dom = isl_pw_aff_domain(isl_pw_aff_copy(Bound));
149560f63b49STobias Grosser       isl_local_space *LS = isl_local_space_from_space(isl_set_get_space(Dom));
149660f63b49STobias Grosser       isl_aff *One = isl_aff_zero_on_domain(LS);
149760f63b49STobias Grosser       One = isl_aff_add_constant_si(One, 1);
149860f63b49STobias Grosser       Bound = isl_pw_aff_add(Bound, isl_pw_aff_alloc(Dom, One));
149960f63b49STobias Grosser       Bound = isl_pw_aff_gist(Bound, S->getContext());
150060f63b49STobias Grosser       PPCGArray.bound[0] = Bound;
150160f63b49STobias Grosser     }
150260f63b49STobias Grosser 
150360f63b49STobias Grosser     for (unsigned i = 1; i < PPCGArray.n_index; ++i) {
150460f63b49STobias Grosser       isl_pw_aff *Bound = Array->getDimensionSizePw(i);
150560f63b49STobias Grosser       auto LS = isl_pw_aff_get_domain_space(Bound);
150660f63b49STobias Grosser       auto Aff = isl_multi_aff_zero(LS);
150760f63b49STobias Grosser       Bound = isl_pw_aff_pullback_multi_aff(Bound, Aff);
150860f63b49STobias Grosser       PPCGArray.bound[i] = Bound;
150960f63b49STobias Grosser     }
151060f63b49STobias Grosser   }
151160f63b49STobias Grosser 
151260f63b49STobias Grosser   /// Create the arrays for @p PPCGProg.
151360f63b49STobias Grosser   ///
151460f63b49STobias Grosser   /// @param PPCGProg The program to compute the arrays for.
151560f63b49STobias Grosser   void createArrays(gpu_prog *PPCGProg) {
151660f63b49STobias Grosser     int i = 0;
151760f63b49STobias Grosser     for (auto &Element : S->arrays()) {
151860f63b49STobias Grosser       ScopArrayInfo *Array = Element.second.get();
151960f63b49STobias Grosser 
152060f63b49STobias Grosser       std::string TypeName;
152160f63b49STobias Grosser       raw_string_ostream OS(TypeName);
152260f63b49STobias Grosser 
152360f63b49STobias Grosser       OS << *Array->getElementType();
152460f63b49STobias Grosser       TypeName = OS.str();
152560f63b49STobias Grosser 
152660f63b49STobias Grosser       gpu_array_info &PPCGArray = PPCGProg->array[i];
152760f63b49STobias Grosser 
152860f63b49STobias Grosser       PPCGArray.space = Array->getSpace();
152960f63b49STobias Grosser       PPCGArray.type = strdup(TypeName.c_str());
153060f63b49STobias Grosser       PPCGArray.size = Array->getElementType()->getPrimitiveSizeInBits() / 8;
153160f63b49STobias Grosser       PPCGArray.name = strdup(Array->getName().c_str());
153260f63b49STobias Grosser       PPCGArray.extent = nullptr;
153360f63b49STobias Grosser       PPCGArray.n_index = Array->getNumberOfDimensions();
153460f63b49STobias Grosser       PPCGArray.bound =
153560f63b49STobias Grosser           isl_alloc_array(S->getIslCtx(), isl_pw_aff *, PPCGArray.n_index);
153660f63b49STobias Grosser       PPCGArray.extent = getExtent(Array);
153760f63b49STobias Grosser       PPCGArray.n_ref = 0;
153860f63b49STobias Grosser       PPCGArray.refs = nullptr;
153960f63b49STobias Grosser       PPCGArray.accessed = true;
154060f63b49STobias Grosser       PPCGArray.read_only_scalar = false;
154160f63b49STobias Grosser       PPCGArray.has_compound_element = false;
154260f63b49STobias Grosser       PPCGArray.local = false;
154360f63b49STobias Grosser       PPCGArray.declare_local = false;
154460f63b49STobias Grosser       PPCGArray.global = false;
154560f63b49STobias Grosser       PPCGArray.linearize = false;
154660f63b49STobias Grosser       PPCGArray.dep_order = nullptr;
154713c78e4dSTobias Grosser       PPCGArray.user = Array;
154860f63b49STobias Grosser 
154960f63b49STobias Grosser       setArrayBounds(PPCGArray, Array);
15502d010dafSTobias Grosser       i++;
1551b9fc860aSTobias Grosser 
1552b9fc860aSTobias Grosser       collect_references(PPCGProg, &PPCGArray);
155360f63b49STobias Grosser     }
155460f63b49STobias Grosser   }
155560f63b49STobias Grosser 
155660f63b49STobias Grosser   /// Create an identity map between the arrays in the scop.
155760f63b49STobias Grosser   ///
155860f63b49STobias Grosser   /// @returns An identity map between the arrays in the scop.
155960f63b49STobias Grosser   isl_union_map *getArrayIdentity() {
156060f63b49STobias Grosser     isl_union_map *Maps = isl_union_map_empty(S->getParamSpace());
156160f63b49STobias Grosser 
156260f63b49STobias Grosser     for (auto &Item : S->arrays()) {
156360f63b49STobias Grosser       ScopArrayInfo *Array = Item.second.get();
156460f63b49STobias Grosser       isl_space *Space = Array->getSpace();
156560f63b49STobias Grosser       Space = isl_space_map_from_set(Space);
156660f63b49STobias Grosser       isl_map *Identity = isl_map_identity(Space);
156760f63b49STobias Grosser       Maps = isl_union_map_add_map(Maps, Identity);
156860f63b49STobias Grosser     }
156960f63b49STobias Grosser 
157060f63b49STobias Grosser     return Maps;
157160f63b49STobias Grosser   }
157260f63b49STobias Grosser 
1573e938517eSTobias Grosser   /// Create a default-initialized PPCG GPU program.
1574e938517eSTobias Grosser   ///
1575e938517eSTobias Grosser   /// @returns A new gpu grogram description.
1576e938517eSTobias Grosser   gpu_prog *createPPCGProg(ppcg_scop *PPCGScop) {
1577e938517eSTobias Grosser 
1578e938517eSTobias Grosser     if (!PPCGScop)
1579e938517eSTobias Grosser       return nullptr;
1580e938517eSTobias Grosser 
1581e938517eSTobias Grosser     auto PPCGProg = isl_calloc_type(S->getIslCtx(), struct gpu_prog);
1582e938517eSTobias Grosser 
1583e938517eSTobias Grosser     PPCGProg->ctx = S->getIslCtx();
1584e938517eSTobias Grosser     PPCGProg->scop = PPCGScop;
1585aef5196fSTobias Grosser     PPCGProg->context = isl_set_copy(PPCGScop->context);
158660f63b49STobias Grosser     PPCGProg->read = isl_union_map_copy(PPCGScop->reads);
158760f63b49STobias Grosser     PPCGProg->may_write = isl_union_map_copy(PPCGScop->may_writes);
158860f63b49STobias Grosser     PPCGProg->must_write = isl_union_map_copy(PPCGScop->must_writes);
158960f63b49STobias Grosser     PPCGProg->tagged_must_kill =
159060f63b49STobias Grosser         isl_union_map_copy(PPCGScop->tagged_must_kills);
159160f63b49STobias Grosser     PPCGProg->to_inner = getArrayIdentity();
159260f63b49STobias Grosser     PPCGProg->to_outer = getArrayIdentity();
159360f63b49STobias Grosser     PPCGProg->may_persist = compute_may_persist(PPCGProg);
1594e938517eSTobias Grosser     PPCGProg->any_to_outer = nullptr;
1595e938517eSTobias Grosser     PPCGProg->array_order = nullptr;
159669b46751STobias Grosser     PPCGProg->n_stmts = std::distance(S->begin(), S->end());
159769b46751STobias Grosser     PPCGProg->stmts = getStatements();
159860f63b49STobias Grosser     PPCGProg->n_array = std::distance(S->array_begin(), S->array_end());
159960f63b49STobias Grosser     PPCGProg->array = isl_calloc_array(S->getIslCtx(), struct gpu_array_info,
160060f63b49STobias Grosser                                        PPCGProg->n_array);
160160f63b49STobias Grosser 
160260f63b49STobias Grosser     createArrays(PPCGProg);
1603e938517eSTobias Grosser 
1604e938517eSTobias Grosser     return PPCGProg;
1605e938517eSTobias Grosser   }
1606e938517eSTobias Grosser 
160769b46751STobias Grosser   struct PrintGPUUserData {
160869b46751STobias Grosser     struct cuda_info *CudaInfo;
160969b46751STobias Grosser     struct gpu_prog *PPCGProg;
161069b46751STobias Grosser     std::vector<ppcg_kernel *> Kernels;
161169b46751STobias Grosser   };
161269b46751STobias Grosser 
161369b46751STobias Grosser   /// Print a user statement node in the host code.
161469b46751STobias Grosser   ///
161569b46751STobias Grosser   /// We use ppcg's printing facilities to print the actual statement and
161669b46751STobias Grosser   /// additionally build up a list of all kernels that are encountered in the
161769b46751STobias Grosser   /// host ast.
161869b46751STobias Grosser   ///
161969b46751STobias Grosser   /// @param P The printer to print to
162069b46751STobias Grosser   /// @param Options The printing options to use
162169b46751STobias Grosser   /// @param Node The node to print
162269b46751STobias Grosser   /// @param User A user pointer to carry additional data. This pointer is
162369b46751STobias Grosser   ///             expected to be of type PrintGPUUserData.
162469b46751STobias Grosser   ///
162569b46751STobias Grosser   /// @returns A printer to which the output has been printed.
162669b46751STobias Grosser   static __isl_give isl_printer *
162769b46751STobias Grosser   printHostUser(__isl_take isl_printer *P,
162869b46751STobias Grosser                 __isl_take isl_ast_print_options *Options,
162969b46751STobias Grosser                 __isl_take isl_ast_node *Node, void *User) {
163069b46751STobias Grosser     auto Data = (struct PrintGPUUserData *)User;
163169b46751STobias Grosser     auto Id = isl_ast_node_get_annotation(Node);
163269b46751STobias Grosser 
163369b46751STobias Grosser     if (Id) {
163420251734STobias Grosser       bool IsUser = !strcmp(isl_id_get_name(Id), "user");
163520251734STobias Grosser 
163620251734STobias Grosser       // If this is a user statement, format it ourselves as ppcg would
163720251734STobias Grosser       // otherwise try to call pet functionality that is not available in
163820251734STobias Grosser       // Polly.
163920251734STobias Grosser       if (IsUser) {
164020251734STobias Grosser         P = isl_printer_start_line(P);
164120251734STobias Grosser         P = isl_printer_print_ast_node(P, Node);
164220251734STobias Grosser         P = isl_printer_end_line(P);
164320251734STobias Grosser         isl_id_free(Id);
164420251734STobias Grosser         isl_ast_print_options_free(Options);
164520251734STobias Grosser         return P;
164620251734STobias Grosser       }
164720251734STobias Grosser 
164869b46751STobias Grosser       auto Kernel = (struct ppcg_kernel *)isl_id_get_user(Id);
164969b46751STobias Grosser       isl_id_free(Id);
165069b46751STobias Grosser       Data->Kernels.push_back(Kernel);
165169b46751STobias Grosser     }
165269b46751STobias Grosser 
165369b46751STobias Grosser     return print_host_user(P, Options, Node, User);
165469b46751STobias Grosser   }
165569b46751STobias Grosser 
165669b46751STobias Grosser   /// Print C code corresponding to the control flow in @p Kernel.
165769b46751STobias Grosser   ///
165869b46751STobias Grosser   /// @param Kernel The kernel to print
165969b46751STobias Grosser   void printKernel(ppcg_kernel *Kernel) {
166069b46751STobias Grosser     auto *P = isl_printer_to_str(S->getIslCtx());
166169b46751STobias Grosser     P = isl_printer_set_output_format(P, ISL_FORMAT_C);
166269b46751STobias Grosser     auto *Options = isl_ast_print_options_alloc(S->getIslCtx());
166369b46751STobias Grosser     P = isl_ast_node_print(Kernel->tree, P, Options);
166469b46751STobias Grosser     char *String = isl_printer_get_str(P);
166569b46751STobias Grosser     printf("%s\n", String);
166669b46751STobias Grosser     free(String);
166769b46751STobias Grosser     isl_printer_free(P);
166869b46751STobias Grosser   }
166969b46751STobias Grosser 
167069b46751STobias Grosser   /// Print C code corresponding to the GPU code described by @p Tree.
167169b46751STobias Grosser   ///
167269b46751STobias Grosser   /// @param Tree An AST describing GPU code
167369b46751STobias Grosser   /// @param PPCGProg The PPCG program from which @Tree has been constructed.
167469b46751STobias Grosser   void printGPUTree(isl_ast_node *Tree, gpu_prog *PPCGProg) {
167569b46751STobias Grosser     auto *P = isl_printer_to_str(S->getIslCtx());
167669b46751STobias Grosser     P = isl_printer_set_output_format(P, ISL_FORMAT_C);
167769b46751STobias Grosser 
167869b46751STobias Grosser     PrintGPUUserData Data;
167969b46751STobias Grosser     Data.PPCGProg = PPCGProg;
168069b46751STobias Grosser 
168169b46751STobias Grosser     auto *Options = isl_ast_print_options_alloc(S->getIslCtx());
168269b46751STobias Grosser     Options =
168369b46751STobias Grosser         isl_ast_print_options_set_print_user(Options, printHostUser, &Data);
168469b46751STobias Grosser     P = isl_ast_node_print(Tree, P, Options);
168569b46751STobias Grosser     char *String = isl_printer_get_str(P);
168669b46751STobias Grosser     printf("# host\n");
168769b46751STobias Grosser     printf("%s\n", String);
168869b46751STobias Grosser     free(String);
168969b46751STobias Grosser     isl_printer_free(P);
169069b46751STobias Grosser 
169169b46751STobias Grosser     for (auto Kernel : Data.Kernels) {
169269b46751STobias Grosser       printf("# kernel%d\n", Kernel->id);
169369b46751STobias Grosser       printKernel(Kernel);
169469b46751STobias Grosser     }
169569b46751STobias Grosser   }
169669b46751STobias Grosser 
1697f384594dSTobias Grosser   // Generate a GPU program using PPCG.
1698f384594dSTobias Grosser   //
1699f384594dSTobias Grosser   // GPU mapping consists of multiple steps:
1700f384594dSTobias Grosser   //
1701f384594dSTobias Grosser   //  1) Compute new schedule for the program.
1702f384594dSTobias Grosser   //  2) Map schedule to GPU (TODO)
1703f384594dSTobias Grosser   //  3) Generate code for new schedule (TODO)
1704f384594dSTobias Grosser   //
1705f384594dSTobias Grosser   // We do not use here the Polly ScheduleOptimizer, as the schedule optimizer
1706f384594dSTobias Grosser   // is mostly CPU specific. Instead, we use PPCG's GPU code generation
1707f384594dSTobias Grosser   // strategy directly from this pass.
1708f384594dSTobias Grosser   gpu_gen *generateGPU(ppcg_scop *PPCGScop, gpu_prog *PPCGProg) {
1709f384594dSTobias Grosser 
1710f384594dSTobias Grosser     auto PPCGGen = isl_calloc_type(S->getIslCtx(), struct gpu_gen);
1711f384594dSTobias Grosser 
1712f384594dSTobias Grosser     PPCGGen->ctx = S->getIslCtx();
1713f384594dSTobias Grosser     PPCGGen->options = PPCGScop->options;
1714f384594dSTobias Grosser     PPCGGen->print = nullptr;
1715f384594dSTobias Grosser     PPCGGen->print_user = nullptr;
171660c60025STobias Grosser     PPCGGen->build_ast_expr = &pollyBuildAstExprForStmt;
1717f384594dSTobias Grosser     PPCGGen->prog = PPCGProg;
1718f384594dSTobias Grosser     PPCGGen->tree = nullptr;
1719f384594dSTobias Grosser     PPCGGen->types.n = 0;
1720f384594dSTobias Grosser     PPCGGen->types.name = nullptr;
1721f384594dSTobias Grosser     PPCGGen->sizes = nullptr;
1722f384594dSTobias Grosser     PPCGGen->used_sizes = nullptr;
1723f384594dSTobias Grosser     PPCGGen->kernel_id = 0;
1724f384594dSTobias Grosser 
1725f384594dSTobias Grosser     // Set scheduling strategy to same strategy PPCG is using.
1726f384594dSTobias Grosser     isl_options_set_schedule_outer_coincidence(PPCGGen->ctx, true);
1727f384594dSTobias Grosser     isl_options_set_schedule_maximize_band_depth(PPCGGen->ctx, true);
17282341fe9eSTobias Grosser     isl_options_set_schedule_whole_component(PPCGGen->ctx, false);
1729f384594dSTobias Grosser 
1730f384594dSTobias Grosser     isl_schedule *Schedule = get_schedule(PPCGGen);
1731f384594dSTobias Grosser 
1732aef5196fSTobias Grosser     int has_permutable = has_any_permutable_node(Schedule);
1733aef5196fSTobias Grosser 
173469b46751STobias Grosser     if (!has_permutable || has_permutable < 0) {
1735aef5196fSTobias Grosser       Schedule = isl_schedule_free(Schedule);
173669b46751STobias Grosser     } else {
1737aef5196fSTobias Grosser       Schedule = map_to_device(PPCGGen, Schedule);
173869b46751STobias Grosser       PPCGGen->tree = generate_code(PPCGGen, isl_schedule_copy(Schedule));
173969b46751STobias Grosser     }
1740aef5196fSTobias Grosser 
1741f384594dSTobias Grosser     if (DumpSchedule) {
1742f384594dSTobias Grosser       isl_printer *P = isl_printer_to_str(S->getIslCtx());
1743f384594dSTobias Grosser       P = isl_printer_set_yaml_style(P, ISL_YAML_STYLE_BLOCK);
1744f384594dSTobias Grosser       P = isl_printer_print_str(P, "Schedule\n");
1745f384594dSTobias Grosser       P = isl_printer_print_str(P, "========\n");
1746f384594dSTobias Grosser       if (Schedule)
1747f384594dSTobias Grosser         P = isl_printer_print_schedule(P, Schedule);
1748f384594dSTobias Grosser       else
1749f384594dSTobias Grosser         P = isl_printer_print_str(P, "No schedule found\n");
1750f384594dSTobias Grosser 
1751f384594dSTobias Grosser       printf("%s\n", isl_printer_get_str(P));
1752f384594dSTobias Grosser       isl_printer_free(P);
1753f384594dSTobias Grosser     }
1754f384594dSTobias Grosser 
175569b46751STobias Grosser     if (DumpCode) {
175669b46751STobias Grosser       printf("Code\n");
175769b46751STobias Grosser       printf("====\n");
175869b46751STobias Grosser       if (PPCGGen->tree)
175969b46751STobias Grosser         printGPUTree(PPCGGen->tree, PPCGProg);
176069b46751STobias Grosser       else
176169b46751STobias Grosser         printf("No code generated\n");
176269b46751STobias Grosser     }
176369b46751STobias Grosser 
1764f384594dSTobias Grosser     isl_schedule_free(Schedule);
1765f384594dSTobias Grosser 
1766f384594dSTobias Grosser     return PPCGGen;
1767f384594dSTobias Grosser   }
1768f384594dSTobias Grosser 
1769f384594dSTobias Grosser   /// Free gpu_gen structure.
1770f384594dSTobias Grosser   ///
1771f384594dSTobias Grosser   /// @param PPCGGen The ppcg_gen object to free.
1772f384594dSTobias Grosser   void freePPCGGen(gpu_gen *PPCGGen) {
1773f384594dSTobias Grosser     isl_ast_node_free(PPCGGen->tree);
1774f384594dSTobias Grosser     isl_union_map_free(PPCGGen->sizes);
1775f384594dSTobias Grosser     isl_union_map_free(PPCGGen->used_sizes);
1776f384594dSTobias Grosser     free(PPCGGen);
1777f384594dSTobias Grosser   }
1778f384594dSTobias Grosser 
1779b307ed4dSTobias Grosser   /// Free the options in the ppcg scop structure.
1780b307ed4dSTobias Grosser   ///
1781b307ed4dSTobias Grosser   /// ppcg is not freeing these options for us. To avoid leaks we do this
1782b307ed4dSTobias Grosser   /// ourselves.
1783b307ed4dSTobias Grosser   ///
1784b307ed4dSTobias Grosser   /// @param PPCGScop The scop referencing the options to free.
1785b307ed4dSTobias Grosser   void freeOptions(ppcg_scop *PPCGScop) {
1786b307ed4dSTobias Grosser     free(PPCGScop->options->debug);
1787b307ed4dSTobias Grosser     PPCGScop->options->debug = nullptr;
1788b307ed4dSTobias Grosser     free(PPCGScop->options);
1789b307ed4dSTobias Grosser     PPCGScop->options = nullptr;
1790b307ed4dSTobias Grosser   }
1791b307ed4dSTobias Grosser 
179238fc0aedSTobias Grosser   /// Generate code for a given GPU AST described by @p Root.
179338fc0aedSTobias Grosser   ///
179432837fe3STobias Grosser   /// @param Root An isl_ast_node pointing to the root of the GPU AST.
179532837fe3STobias Grosser   /// @param Prog The GPU Program to generate code for.
179632837fe3STobias Grosser   void generateCode(__isl_take isl_ast_node *Root, gpu_prog *Prog) {
179738fc0aedSTobias Grosser     ScopAnnotator Annotator;
179838fc0aedSTobias Grosser     Annotator.buildAliasScopes(*S);
179938fc0aedSTobias Grosser 
180038fc0aedSTobias Grosser     Region *R = &S->getRegion();
180138fc0aedSTobias Grosser 
180238fc0aedSTobias Grosser     simplifyRegion(R, DT, LI, RI);
180338fc0aedSTobias Grosser 
180438fc0aedSTobias Grosser     BasicBlock *EnteringBB = R->getEnteringBlock();
180538fc0aedSTobias Grosser 
180638fc0aedSTobias Grosser     PollyIRBuilder Builder = createPollyIRBuilder(EnteringBB, Annotator);
180738fc0aedSTobias Grosser 
180832837fe3STobias Grosser     GPUNodeBuilder NodeBuilder(Builder, Annotator, this, *DL, *LI, *SE, *DT, *S,
180932837fe3STobias Grosser                                Prog);
181038fc0aedSTobias Grosser 
181138fc0aedSTobias Grosser     // Only build the run-time condition and parameters _after_ having
181238fc0aedSTobias Grosser     // introduced the conditional branch. This is important as the conditional
181338fc0aedSTobias Grosser     // branch will guard the original scop from new induction variables that
181438fc0aedSTobias Grosser     // the SCEVExpander may introduce while code generating the parameters and
181538fc0aedSTobias Grosser     // which may introduce scalar dependences that prevent us from correctly
181638fc0aedSTobias Grosser     // code generating this scop.
181738fc0aedSTobias Grosser     BasicBlock *StartBlock =
181838fc0aedSTobias Grosser         executeScopConditionally(*S, this, Builder.getTrue());
181938fc0aedSTobias Grosser 
182038fc0aedSTobias Grosser     // TODO: Handle LICM
182138fc0aedSTobias Grosser     // TODO: Verify run-time checks
182238fc0aedSTobias Grosser     auto SplitBlock = StartBlock->getSinglePredecessor();
182338fc0aedSTobias Grosser     Builder.SetInsertPoint(SplitBlock->getTerminator());
182438fc0aedSTobias Grosser     NodeBuilder.addParameters(S->getContext());
182538fc0aedSTobias Grosser     Builder.SetInsertPoint(&*StartBlock->begin());
1826fa7b0802STobias Grosser 
1827fa7b0802STobias Grosser     NodeBuilder.initializeAfterRTH();
182838fc0aedSTobias Grosser     NodeBuilder.create(Root);
18298ed5e599STobias Grosser     NodeBuilder.finalize();
183038fc0aedSTobias Grosser   }
183138fc0aedSTobias Grosser 
1832e938517eSTobias Grosser   bool runOnScop(Scop &CurrentScop) override {
1833e938517eSTobias Grosser     S = &CurrentScop;
183438fc0aedSTobias Grosser     LI = &getAnalysis<LoopInfoWrapperPass>().getLoopInfo();
183538fc0aedSTobias Grosser     DT = &getAnalysis<DominatorTreeWrapperPass>().getDomTree();
183638fc0aedSTobias Grosser     SE = &getAnalysis<ScalarEvolutionWrapperPass>().getSE();
183738fc0aedSTobias Grosser     DL = &S->getRegion().getEntry()->getParent()->getParent()->getDataLayout();
183838fc0aedSTobias Grosser     RI = &getAnalysis<RegionInfoPass>().getRegionInfo();
1839e938517eSTobias Grosser 
18402d58a64eSTobias Grosser     // We currently do not support scops with invariant loads.
18412d58a64eSTobias Grosser     if (S->hasInvariantAccesses())
18422d58a64eSTobias Grosser       return false;
18432d58a64eSTobias Grosser 
1844e938517eSTobias Grosser     auto PPCGScop = createPPCGScop();
1845e938517eSTobias Grosser     auto PPCGProg = createPPCGProg(PPCGScop);
1846f384594dSTobias Grosser     auto PPCGGen = generateGPU(PPCGScop, PPCGProg);
184738fc0aedSTobias Grosser 
184838fc0aedSTobias Grosser     if (PPCGGen->tree)
184932837fe3STobias Grosser       generateCode(isl_ast_node_copy(PPCGGen->tree), PPCGProg);
185038fc0aedSTobias Grosser 
1851b307ed4dSTobias Grosser     freeOptions(PPCGScop);
1852f384594dSTobias Grosser     freePPCGGen(PPCGGen);
1853e938517eSTobias Grosser     gpu_prog_free(PPCGProg);
1854e938517eSTobias Grosser     ppcg_scop_free(PPCGScop);
1855e938517eSTobias Grosser 
1856e938517eSTobias Grosser     return true;
1857e938517eSTobias Grosser   }
18589dfe4e7cSTobias Grosser 
18599dfe4e7cSTobias Grosser   void printScop(raw_ostream &, Scop &) const override {}
18609dfe4e7cSTobias Grosser 
18619dfe4e7cSTobias Grosser   void getAnalysisUsage(AnalysisUsage &AU) const override {
18629dfe4e7cSTobias Grosser     AU.addRequired<DominatorTreeWrapperPass>();
18639dfe4e7cSTobias Grosser     AU.addRequired<RegionInfoPass>();
18649dfe4e7cSTobias Grosser     AU.addRequired<ScalarEvolutionWrapperPass>();
18659dfe4e7cSTobias Grosser     AU.addRequired<ScopDetection>();
18669dfe4e7cSTobias Grosser     AU.addRequired<ScopInfoRegionPass>();
18679dfe4e7cSTobias Grosser     AU.addRequired<LoopInfoWrapperPass>();
18689dfe4e7cSTobias Grosser 
18699dfe4e7cSTobias Grosser     AU.addPreserved<AAResultsWrapperPass>();
18709dfe4e7cSTobias Grosser     AU.addPreserved<BasicAAWrapperPass>();
18719dfe4e7cSTobias Grosser     AU.addPreserved<LoopInfoWrapperPass>();
18729dfe4e7cSTobias Grosser     AU.addPreserved<DominatorTreeWrapperPass>();
18739dfe4e7cSTobias Grosser     AU.addPreserved<GlobalsAAWrapperPass>();
18749dfe4e7cSTobias Grosser     AU.addPreserved<PostDominatorTreeWrapperPass>();
18759dfe4e7cSTobias Grosser     AU.addPreserved<ScopDetection>();
18769dfe4e7cSTobias Grosser     AU.addPreserved<ScalarEvolutionWrapperPass>();
18779dfe4e7cSTobias Grosser     AU.addPreserved<SCEVAAWrapperPass>();
18789dfe4e7cSTobias Grosser 
18799dfe4e7cSTobias Grosser     // FIXME: We do not yet add regions for the newly generated code to the
18809dfe4e7cSTobias Grosser     //        region tree.
18819dfe4e7cSTobias Grosser     AU.addPreserved<RegionInfoPass>();
18829dfe4e7cSTobias Grosser     AU.addPreserved<ScopInfoRegionPass>();
18839dfe4e7cSTobias Grosser   }
18849dfe4e7cSTobias Grosser };
18859dfe4e7cSTobias Grosser }
18869dfe4e7cSTobias Grosser 
18879dfe4e7cSTobias Grosser char PPCGCodeGeneration::ID = 1;
18889dfe4e7cSTobias Grosser 
18899dfe4e7cSTobias Grosser Pass *polly::createPPCGCodeGenerationPass() { return new PPCGCodeGeneration(); }
18909dfe4e7cSTobias Grosser 
18919dfe4e7cSTobias Grosser INITIALIZE_PASS_BEGIN(PPCGCodeGeneration, "polly-codegen-ppcg",
18929dfe4e7cSTobias Grosser                       "Polly - Apply PPCG translation to SCOP", false, false)
18939dfe4e7cSTobias Grosser INITIALIZE_PASS_DEPENDENCY(DependenceInfo);
18949dfe4e7cSTobias Grosser INITIALIZE_PASS_DEPENDENCY(DominatorTreeWrapperPass);
18959dfe4e7cSTobias Grosser INITIALIZE_PASS_DEPENDENCY(LoopInfoWrapperPass);
18969dfe4e7cSTobias Grosser INITIALIZE_PASS_DEPENDENCY(RegionInfoPass);
18979dfe4e7cSTobias Grosser INITIALIZE_PASS_DEPENDENCY(ScalarEvolutionWrapperPass);
18989dfe4e7cSTobias Grosser INITIALIZE_PASS_DEPENDENCY(ScopDetection);
18999dfe4e7cSTobias Grosser INITIALIZE_PASS_END(PPCGCodeGeneration, "polly-codegen-ppcg",
19009dfe4e7cSTobias Grosser                     "Polly - Apply PPCG translation to SCOP", false, false)
1901