コード例 #1
0
def objective(fields, solver):
    U, T, p = fields
    mesh = solver.mesh.symMesh

    def _meshArgs(start=0):
        return [x[start] for x in mesh.getTensor()]

    patch = mesh.boundary['obstacle']
    startFace, nFaces = patch['startFace'], patch['nFaces']
    meshArgs = _meshArgs(startFace)
    test = tensor.Zeros((1, 1))
    test = tensor.Kernel(objectiveTest)(nFaces, (test, ))(U, T, p, *meshArgs)

    inputs = (test, )
    outputs = tuple([tensor.Zeros(x.shape) for x in inputs])
    (test, ) = tensor.ExternalFunctionOp('mpi_allreduce', inputs,
                                         outputs).outputs
    return test
コード例 #2
0
ファイル: vane.py プロジェクト: qiqi/adFVM
def objective(fields, solver):
    U, T, p = fields
    mesh = solver.mesh.symMesh

    def _meshArgs(start=0):
        return [x[start] for x in mesh.getTensor()]

    nPlaneCells, cells, areas = [x[0] for x in solver.extraArgs[:3]]
    w = tensor.Zeros((1, 1))
    w = tensor.Kernel(objectivePressureLossWeighting)(nPlaneCells,
                                                      (w, ))(U,
                                                             T,
                                                             p,
                                                             cells,
                                                             areas,
                                                             solver=solver)

    weights = [x[0] for x in solver.extraArgs[3:]]
    w2 = tensor.Zeros((1, 1))
    _heatTransferWeighting = tensor.Kernel(objectiveHeatTransferWeighting)
    for index, patchID in enumerate(patches):
        patch = mesh.boundary[patchID]
        startFace, nFaces = patch['startFace'], patch['nFaces']
        meshArgs = _meshArgs(startFace)
        weight = weights[index]
        w2 = _heatTransferWeighting(nFaces, (w2, ))(weight, *meshArgs)

    inputs = (w, w2)
    outputs = tuple([tensor.Zeros(x.shape) for x in inputs])
    w, w2 = tensor.ExternalFunctionOp('mpi_allreduce', inputs, outputs).outputs

    pl = tensor.Zeros((1, 1))
    pl = tensor.Kernel(objectivePressureLoss)(nPlaneCells,
                                              (pl, ))(U,
                                                      T,
                                                      p,
                                                      cells,
                                                      areas,
                                                      w,
                                                      solver=solver)

    _heatTransfer = tensor.Kernel(objectiveHeatTransfer)
    ht = tensor.Zeros((1, 1))
    for index, patchID in enumerate(patches):
        patch = mesh.boundary[patchID]
        startFace, nFaces = patch['startFace'], patch['nFaces']
        meshArgs = _meshArgs(startFace)
        weight = weights[index]
        ht = _heatTransfer(nFaces, (ht, ))(U,
                                           T,
                                           p,
                                           weight,
                                           w2,
                                           *meshArgs,
                                           solver=solver)

    k = solver.mu(300) * solver.Cp / solver.Pr
    a = 0.4
    #b = -0.71e-3/(120*k)/2000.
    b = 0.

    # MPI ALLREDUCE
    #if not config.gpu:
    #    inputs = (pl, w, ht, w2)
    #    outputs = tuple([tensor.Zeros(x.shape) for x in inputs])
    #    pl, w, ht, w2 = tensor.ExternalFunctionOp('mpi_allreduce', inputs, outputs).outputs
    inputs = (pl, ht)
    outputs = tuple([tensor.Zeros(x.shape) for x in inputs])
    pl, ht = tensor.ExternalFunctionOp('mpi_allreduce', inputs,
                                       outputs).outputs

    # then elemwise
    def _combine(pl, ht):
        pl = pl.scalar()
        ht = ht.scalar()
        obj = pl
        obj2 = ht
        return a * obj + b * obj2

    return tensor.Kernel(_combine)(1)(pl, ht)