Home My Page Projects Code Snippets Project Openings diderot
Summary Activity Tracker Tasks SCM

SCM Repository

[diderot] Diff of /branches/pure-cfg/src/compiler/cl-target/cl-target.sml
ViewVC logotype

Diff of /branches/pure-cfg/src/compiler/cl-target/cl-target.sml

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1397, Mon Jun 27 16:04:43 2011 UTC revision 1398, Wed Jun 29 17:20:05 2011 UTC
# Line 400  Line 400 
400                  List.rev (List.map (fn x => (targetTy x, #var x)) (!state)),                  List.rev (List.map (fn x => (targetTy x, #var x)) (!state)),
401                  tyName)                  tyName)
402    
         fun genStrandCopy(Strand{tyName,name,state,...}) = let  
               val params = [  
                     CL.PARAM(["__global"], CL.T_Ptr(CL.T_Named tyName), "selfIn"),  
                     CL.PARAM(["__global"], CL.T_Ptr(CL.T_Named tyName), "selfOut")  
                   ]  
               val assignStms = List.rev(List.map(fn x => CL.mkAssign(CL.mkIndirect(CL.E_Var "selfOut", #var x),                                                                                                                          CL.mkIndirect(CL.E_Var "selfIn", #var x))) (!state))  
               in  
                 CL.D_Func([""], CL.voidTy, RN.strandCopy name, params,CL.mkBlock(assignStms))  
               end  
   
403        (* generates the opencl buffers for the image data *)        (* generates the opencl buffers for the image data *)
404          fun getGlobalDataBuffers (globals, contextVar, errVar) = let          fun getGlobalDataBuffers (globals, imgGlobals, contextVar, errVar) = let
405                val globalBuffErr = "error creating OpenCL global buffer"                val globalBuffErr = "error creating OpenCL global buffer"
406                fun errorFn msg = CL.mkIfThen(CL.mkBinOp(CL.E_Var errVar, CL.#!=, CL.E_Var "CL_SUCCESS"),                fun errorFn msg = CL.mkIfThen(CL.mkBinOp(CL.E_Var errVar, CL.#!=, CL.E_Var "CL_SUCCESS"),
407                      CL.mkBlock([CL.mkCall("fprintf",[CL.E_Var "stderr", CL.E_Str msg]),                      CL.mkBlock([CL.mkCall("fprintf",[CL.E_Var "stderr", CL.E_Str msg]),
408                      CL.mkCall("exit",[CL.mkInt 1])]))                      CL.mkCall("exit",[CL.mkInt 1])]))
409                  val shadowTypeDecl = CL.mkDecl(CL.T_Named(RN.shadowGlobalsTy),RN.shadowGlaobalsName,NONE)
410                  val globalImagesToShadowStms = List.map (fn (var,nDims) =>
411                        CL.mkCall((RN.shadowImageFunc nDims), [
412                             CL.mkVar contextVar,
413                             CL.mkUnOp(CL.%&,CL.mkSelect(CL.mkVar(RN.shadowGlaobalsName),var)),
414                             CL.mkIndirect(CL.mkVar(RN.globalsVarName),var)
415                             ])) imgGlobals
416                  (* Converts only the non-image-field types into their corresponding global shadow types *)
417                  fun convertToShadow ((global:mirror_var)::rest,(imgGlobal,nDim)::restImages) =
418                         if (#var global) = imgGlobal then
419                            convertToShadow(rest,restImages)
420                        else
421                            CL.mkAssign(CL.mkSelect(CL.mkVar(RN.shadowGlaobalsName),#var global),
422                                    CL.mkIndirect(CL.mkVar(RN.globalsVarName), #var global))::
423                                        convertToShadow(rest,(imgGlobal,nDim)::restImages)
424                    | convertToShadow ([], _) = []
425                    | convertToShadow ((global:mirror_var)::rest, []) =
426                          CL.mkAssign(CL.mkSelect(CL.mkVar(RN.shadowGlaobalsName),#var global),
427                                    CL.mkIndirect(CL.mkVar(RN.globalsVarName), #var global))::
428                                        convertToShadow(rest,[])
429    
430                  val globalToShadowStms = convertToShadow(globals,imgGlobals)
431                val globalBufferDecl = CL.mkDecl(clMemoryTy,concat[RN.globalsVarName,"_cl"],NONE)                val globalBufferDecl = CL.mkDecl(clMemoryTy,concat[RN.globalsVarName,"_cl"],NONE)
432                val globalBuffer = CL.mkAssign(CL.mkVar(concat[RN.globalsVarName,"_cl"]),                val globalBuffer = CL.mkAssign(CL.mkVar(concat[RN.globalsVarName,"_cl"]),
433                      CL.mkApply("clCreateBuffer", [                      CL.mkApply("clCreateBuffer", [
434                          CL.mkVar contextVar,                          CL.mkVar contextVar,
435                          CL.mkVar "CL_MEM_READ_WRITE | CL_MEM_COPY_HOST_PTR",                          CL.mkVar "CL_MEM_READ_ONLY | CL_MEM_COPY_HOST_PTR",
436                          CL.mkSizeof(CL.T_Named RN.globalsTy),                          CL.mkSizeof(CL.T_Named RN.shadowGlobalsTy),
437                          CL.mkVar RN.globalsVarName,                          CL.mkUnOp(CL.%&,CL.mkVar RN.shadowGlaobalsName),
438                          CL.mkUnOp(CL.%&,CL.mkVar errVar)                          CL.mkUnOp(CL.%&,CL.mkVar errVar)
439                        ]))                        ]))
440                fun genDataBuffers ([],_,_,_) = []                fun genDataBuffers ([],_,_,_) = []
441                  | genDataBuffers ((var,nDims)::globals, contextVar, errVar,errFn) = let                  | genDataBuffers ((var,nDims)::globals, contextVar, errVar,errFn) = let
442                      val hostVar = CL.mkIndirect(CL.mkVar RN.globalsVarName, var)                      val hostVar = CL.mkIndirect(CL.mkVar RN.globalsVarName, var)
443  (* FIXME: use CL constructors to build expressions (not strings) *)                      val size = CL.mkIndirect(hostVar, "dataSzb")
                     fun sizeExp i = CL.mkSubscript(CL.mkIndirect(hostVar, "size"), CL.mkInt i)  
 (* FIXME: there is no reason that images have to be restricted to float elements! *)  
                     val size = CL.mkBinOp(CL.mkSizeof(CL.float), CL.#*, sizeExp 0)  
                     val size = if (nDims > 1)  
                           then CL.mkBinOp(size, CL.#*, sizeExp 1)  
                           else size  
                     val size = if (nDims > 2)  
                           then CL.mkBinOp(size, CL.#*, sizeExp 2)  
                           else size  
444                      in                      in
                       CL.mkDecl(clMemoryTy, RN.addBufferSuffix var ,NONE)::  
445                        CL.mkDecl(clMemoryTy, RN.addBufferSuffixData var ,NONE)::                        CL.mkDecl(clMemoryTy, RN.addBufferSuffixData var ,NONE)::
                       CL.mkAssign(CL.mkVar(RN.addBufferSuffix var),  
                         CL.mkApply("clCreateBuffer", [  
                             CL.mkVar contextVar,  
                             CL.mkVar "CL_MEM_COPY_HOST_PTR",  
                             CL.mkSizeof(CL.T_Named(RN.imageTy nDims)),  
                             hostVar,  
                             CL.mkUnOp(CL.%&,CL.mkVar errVar)  
                           ])) ::  
                       errFn(concat["error in creating ",RN.addBufferSuffix var, " global buffer"]) ::  
446                        CL.mkAssign(CL.mkVar(RN.addBufferSuffixData var),                        CL.mkAssign(CL.mkVar(RN.addBufferSuffixData var),
447                          CL.mkApply("clCreateBuffer", [                          CL.mkApply("clCreateBuffer", [
448                              CL.mkVar contextVar,                              CL.mkVar contextVar,
449                              CL.mkVar "CL_MEM_COPY_HOST_PTR",                              CL.mkVar "CL_MEM_READ_ONLY | CL_MEM_COPY_HOST_PTR",
450                              size,                              size,
451                              CL.mkIndirect(hostVar, "data"),                              CL.mkIndirect(hostVar, "data"),
452                              CL.mkUnOp(CL.%&,CL.mkVar errVar)                              CL.mkUnOp(CL.%&,CL.mkVar errVar)
# Line 462  Line 455 
455                          genDataBuffers(globals,contextVar,errVar,errFn)                          genDataBuffers(globals,contextVar,errVar,errFn)
456                      end                      end
457                in                in
458                  globalBufferDecl                  [shadowTypeDecl] @ globalImagesToShadowStms @ globalToShadowStms
459                  :: globalBuffer                  @ [globalBufferDecl, globalBuffer,errorFn(globalBuffErr)]
460                  :: errorFn(globalBuffErr)                  @ genDataBuffers(imgGlobals,contextVar,errVar,errorFn)
                 :: genDataBuffers(globals,contextVar,errVar,errorFn)  
461                end                end
462    
463        (* generates the kernel arguments for the image data *)        (* generates the kernel arguments for the image data *)
# Line 474  Line 466 
466                fun errorFn msg = CL.mkIfThen(CL.mkBinOp(CL.E_Var errVar, CL.#!=, CL.E_Var "CL_SUCCESS"),                fun errorFn msg = CL.mkIfThen(CL.mkBinOp(CL.E_Var errVar, CL.#!=, CL.E_Var "CL_SUCCESS"),
467                      CL.mkBlock([CL.mkCall("fprintf",[CL.E_Var "stderr", CL.E_Str msg]),                      CL.mkBlock([CL.mkCall("fprintf",[CL.E_Var "stderr", CL.E_Str msg]),
468                      CL.mkCall("exit",[CL.mkInt 1])]))                      CL.mkCall("exit",[CL.mkInt 1])]))
469                val globalArgument = CL.mkExpStm(CL.mkAssignOp(CL.mkVar errVar,CL.|=,                val globalArgument = CL.mkExpStm(CL.mkAssignOp(CL.mkVar errVar,CL.&=,
470                      CL.mkApply("clSetKernelArg",                      CL.mkApply("clSetKernelArg",
471                        [CL.mkVar kernelVar,                        [CL.mkVar kernelVar,
472                         CL.mkPostOp(CL.E_Var count, CL.^++),                         CL.mkPostOp(CL.E_Var count, CL.^++),
# Line 487  Line 479 
479                          [CL.mkVar kernelVar,                          [CL.mkVar kernelVar,
480                           CL.mkPostOp(CL.E_Var count, CL.^++),                           CL.mkPostOp(CL.E_Var count, CL.^++),
481                           CL.mkApply("sizeof",[CL.mkVar "cl_mem"]),                           CL.mkApply("sizeof",[CL.mkVar "cl_mem"]),
                          CL.mkUnOp(CL.%&,CL.mkVar(RN.addBufferSuffix var))]))) ::  
                          errFn(concat["error in creating ",RN.addBufferSuffix var, " argument"]) ::  
                     CL.mkExpStm(CL.mkAssignOp(CL.mkVar errVar,CL.$=,  
                       CL.mkApply("clSetKernelArg",  
                         [CL.mkVar kernelVar,  
                          CL.mkPostOp(CL.E_Var count, CL.^++),  
                          CL.mkApply("sizeof",[CL.mkVar "cl_mem"]),  
482                           CL.mkUnOp(CL.%&,CL.mkVar(RN.addBufferSuffixData var))]))) ::                           CL.mkUnOp(CL.%&,CL.mkVar(RN.addBufferSuffixData var))]))) ::
483                           errFn(concat["error in creating ",RN.addBufferSuffixData var, " argument"]) ::                           errFn(concat["error in creating ",RN.addBufferSuffixData var, " argument"]) ::
484                      genDataArguments (globals,count,kernelVar,errVar,errFn)                      genDataArguments (globals,count,kernelVar,errVar,errFn)
# Line 502  Line 487 
487                end                end
488    
489        (* generates the globals buffers and arguments function *)        (* generates the globals buffers and arguments function *)
490          fun genGlobalBuffersArgs imgGlobals = let          fun genGlobalBuffersArgs (globals,imgGlobals) = let
491              (* Delcare opencl setup objects *)              (* Delcare opencl setup objects *)
492                val errVar = "err"                val errVar = "err"
493                val imgDataSizeVar = "image_dataSize"                val imgDataSizeVar = "image_dataSize"
# Line 512  Line 497 
497                        CL.PARAM([],CL.T_Named("cl_command_queue"), "cmdQ"),                        CL.PARAM([],CL.T_Named("cl_command_queue"), "cmdQ"),
498                        CL.PARAM([],CL.T_Named("int"), "argStart")                        CL.PARAM([],CL.T_Named("int"), "argStart")
499                      ]                      ]
500                val clGlobalBuffers = getGlobalDataBuffers(!imgGlobals, "context", errVar)                val clGlobalBuffers = getGlobalDataBuffers(globals,!imgGlobals, "context", errVar)
501                val clGlobalArguments = genGlobalArguments(!imgGlobals, "argStart", "kernel", errVar)                val clGlobalArguments = genGlobalArguments(!imgGlobals, "argStart", "kernel", errVar)
502              (* Body put all the statments together *)              (* Body put all the statments together *)
503                val body = CL.mkDecl(clIntTy, errVar, SOME(CL.I_Exp(CL.mkInt 0)))                val body = CL.mkDecl(clIntTy, errVar, SOME(CL.I_Exp(CL.mkInt 0)))
# Line 523  Line 508 
508    
509        (* generate the data and global parameters *)        (* generate the data and global parameters *)
510          fun genKeneralGlobalParams ((name,tyname)::rest) =          fun genKeneralGlobalParams ((name,tyname)::rest) =
               globalParam (CL.T_Ptr(CL.T_Named (RN.imageTy tyname)), RN.addBufferSuffix name) ::  
511                globalParam (CL.T_Ptr(CL.voidTy), RN.addBufferSuffixData name) ::                globalParam (CL.T_Ptr(CL.voidTy), RN.addBufferSuffixData name) ::
512                genKeneralGlobalParams rest                genKeneralGlobalParams rest
513            | genKeneralGlobalParams [] = []            | genKeneralGlobalParams [] = []
514    
       (*generate code for intilizing kernel global data *)  
 (* FIXME: should use List.map here *)  
         fun initGlobalImages ((name, tyname)::rest) =  
               CL.mkAssign(  
                 CL.mkIndirect(CL.E_Var RN.globalsVarName, name),  
                 CL.mkVar (RN.addBufferSuffix name)) ::  
               CL.mkAssign(  
                 CL.mkIndirect(CL.mkIndirect(CL.E_Var RN.globalsVarName, name), "data"),  
                 CL.mkVar (RN.addBufferSuffixData name)) ::  
               initGlobalImages rest  
           | initGlobalImages [] = []  
   
515        (* generate the main kernel function for the .cl file *)        (* generate the main kernel function for the .cl file *)
516          fun genKernelFun (strand, nDims, globals, imgGlobals) = let          fun genKernelFun (strand, nDims, globals, imgGlobals) = let
517                val Strand{name, tyName, state, output, code,...} = strand                val Strand{name, tyName, state, output, code,...} = strand
# Line 572  Line 544 
544                        CL.mkDecl(CL.T_Ptr(CL.T_Named (concat["__global ",tyName])), outState, NONE),                        CL.mkDecl(CL.T_Ptr(CL.T_Named (concat["__global ",tyName])), outState, NONE),
545                        CL.mkDecl(CL.T_Ptr(CL.T_Named (concat["__global ",tyName])), tempVar, NONE)                        CL.mkDecl(CL.T_Ptr(CL.T_Named (concat["__global ",tyName])), tempVar, NONE)
546                      ]                      ]
547                val barrierCode = CL.mkCall(RN.strandCopy name, [CL.E_Var outState, CL.E_Var inState])                val imageDataDecl = CL.mkDecl(CL.T_Named(RN.imageDataType),RN.globalImageDataName,NONE)
548                  val imageDataStms = List.map (fn (x,_) =>
549                      CL.mkAssign(CL.mkSelect(CL.mkVar(RN.globalImageDataName),RN.imageDataName x),
550                                  CL.mkVar(RN.addBufferSuffixData x))) (!imgGlobals)
551                  val barrierCode = CL.mkIfThen(CL.mkBinOp(CL.E_Var "status",CL.#==,CL.E_Var "DIDEROT_ACTIVE"),
552                                     CL.mkBlock ([CL.mkAssign(CL.E_Var tempVar, CL.E_Var inState),
553                                     CL.mkAssign(CL.E_Var inState, CL.E_Var outState),
554                                     CL.mkAssign(CL.E_Var outState, CL.E_Var tempVar)]))
555                val barrierStm = CL.mkCall("barrier",[CL.E_Var "CLK_LOCAL_MEM_FENCE"])                val barrierStm = CL.mkCall("barrier",[CL.E_Var "CLK_LOCAL_MEM_FENCE"])
556                val index = if nDims = 1 then                val index = if nDims = 1 then
557                          CL.mkStr "x"                          CL.mkStr "x"
# Line 589  Line 568 
568                                                                          CL.E_Var "status")                                                                          CL.E_Var "status")
569                val status = CL.mkDecl(CL.intTy, "status", SOME(CL.I_Exp(CL.mkSubscript(CL.mkVar "strandStatus",index))))                val status = CL.mkDecl(CL.intTy, "status", SOME(CL.I_Exp(CL.mkSubscript(CL.mkVar "strandStatus",index))))
570                val strandInitStm = CL.mkCall(RN.strandInit name, [                val strandInitStm = CL.mkCall(RN.strandInit name, [
571                        CL.E_Var RN.globalsVarName,                        CL.mkVar RN.globalsVarName,
572                        CL.E_Var outState,                        CL.mkVar inState,
573                        CL.E_Var "x",                        CL.mkVar "x",
574  (* FIXME: if nDims = 1, then "y" is not defined! the arguments to this call should really come from  (* FIXME: if nDims = 1, then "y" is not defined! the arguments to this call should really come from
575   * the initially code!   * the initially code!
576   *)   *)
577                        CL.E_Var "y"])                        CL.mkVar "y"])
578                val local_vars = thread_ids                val local_vars = thread_ids
579                      @ initGlobalImages(!imgGlobals)                      @ [imageDataDecl]
580                        @ imageDataStms
581                      @ strandDecl                      @ strandDecl
582                      @ strandObjects                      @ strandObjects
583                      @ [strandInitStm,status]                      @ [strandInitStm,status]
584                val while_exp = CL.mkBinOp(CL.mkVar "status",CL.#==, CL.mkVar RN.kActive)                val while_exp = CL.mkBinOp(CL.mkVar "status",CL.#==, CL.mkVar RN.kActive)
585                val whileBody = CL.mkBlock ([barrierCode,barrierStm] @ [                val whileBody = CL.mkBlock ([
586                        CL.mkAssign(CL.mkVar "status",                        CL.mkAssign(CL.mkVar "status",
587                          CL.mkApply(RN.strandUpdate name,                          CL.mkApply(RN.strandUpdate name,
588                            [CL.mkVar inState, CL.mkVar outState,CL.E_Var RN.globalsVarName]))] )                            [CL.mkVar inState,
589                               CL.mkVar outState,
590                               CL.mkVar RN.globalsVarName,
591                               CL.mkVar RN.globalImageDataName]))] @ [barrierCode,barrierStm] )
592                val whileBlock = [CL.mkWhile(while_exp, whileBody)]                val whileBlock = [CL.mkWhile(while_exp, whileBody)]
593                val body = CL.mkBlock(local_vars @ whileBlock @ [stabalizeStm])                val body = CL.mkBlock(local_vars @ whileBlock @ [stabalizeStm])
594                in                in
# Line 619  Line 602 
602                  CL.D_StructDef(globs, tyName)                  CL.D_StructDef(globs, tyName)
603                end                end
604    
605          (* generate a global structure type definition from the image data of the image globals *)
606            fun genImageDataStruct (imgGlobals, tyName) = let
607                  val globs = List.map (fn (x,_) => (CL.T_Ptr(CL.imageDataTy "__global"),(RN.imageDataName x))) imgGlobals
608                  in
609                    CL.D_StructDef(globs, tyName)
610                  end
611    
612          fun genGlobals (declFn, targetTy, globals) = let          fun genGlobals (declFn, targetTy, globals) = let
613                fun doVar (x : mirror_var) = declFn (CL.D_Var([], targetTy x, #var x, NONE))                fun doVar (x : mirror_var) = declFn (CL.D_Var([], targetTy x, #var x, NONE))
614                in                in
# Line 684  Line 674 
674                      "#include \"Diderot/cl-diderot.h\""                      "#include \"Diderot/cl-diderot.h\""
675                    ]));                    ]));
676                  clppDecl (genGlobalStruct (#gpuTy, !globals, RN.globalsTy));                  clppDecl (genGlobalStruct (#gpuTy, !globals, RN.globalsTy));
677                    clppDecl (genImageDataStruct(!imgGlobals,RN.imageDataType));
678                  clppDecl (genStrandTyDef(#gpuTy, strand));                  clppDecl (genStrandTyDef(#gpuTy, strand));
679                  clppDecl  (!init_code);                  clppDecl  (!init_code);
                 clppDecl  (genStrandCopy(strand));  
680                  List.app clppDecl (!code);                  List.app clppDecl (!code);
681                  clppDecl (genKernelFun (strand, !numDims, globals, imgGlobals));                  clppDecl (genKernelFun (strand, !numDims, globals, imgGlobals));
682                (* Generate the Host C file *)                (* Generate the Host C file *)
# Line 706  Line 696 
696                  cppDecl (genStrandTyDef (#hostTy, strand));                  cppDecl (genStrandTyDef (#hostTy, strand));
697                  cppDecl (genStrandPrint strand);                  cppDecl (genStrandPrint strand);
698                  List.app cppDecl (List.rev (!topDecls));                  List.app cppDecl (List.rev (!topDecls));
699                  cppDecl (genGlobalBuffersArgs imgGlobals);                  cppDecl (genGlobalBuffersArgs (!globals,imgGlobals));
700                  List.app (fn strand => cppDecl (genStrandDesc strand)) strands;                  List.app (fn strand => cppDecl (genStrandDesc strand)) strands;
701                  genStrandTable (cppDecl, strands);                  genStrandTable (cppDecl, strands);
702                  cppDecl (!initially);                  cppDecl (!initially);
# Line 785  Line 775 
775                val params = [                val params = [
776                        globalParam (CL.T_Ptr(CL.T_Named tyName), "selfIn"),                        globalParam (CL.T_Ptr(CL.T_Named tyName), "selfIn"),
777                        globalParam (CL.T_Ptr(CL.T_Named tyName), "selfOut"),                        globalParam (CL.T_Ptr(CL.T_Named tyName), "selfOut"),
778                        globalParam (CL.T_Ptr(CL.T_Named (RN.globalsTy)), RN.globalsVarName)                        globalParam (CL.T_Ptr(CL.T_Named (RN.globalsTy)), RN.globalsVarName),
779                          CL.PARAM([],CL.T_Named(RN.imageDataType),RN.globalImageDataName)
780                      ]                      ]
781                val methFn = CL.D_Func([], CL.int32, fName, params, body)                val methFn = CL.D_Func([], CL.int32, fName, params, body)
782                in                in

Legend:
Removed from v.1397  
changed lines
  Added in v.1398

root@smlnj-gforge.cs.uchicago.edu
ViewVC Help
Powered by ViewVC 1.0.0