def __init__(self, is_lift, interiors, boundaries,
quadrature_tag):
from hedge.optemplate.tools import get_flux_dependencies
self.is_lift = is_lift
self.interiors = tuple(interiors)
self.boundaries = tuple(boundaries)
self.quadrature_tag = quadrature_tag
from pytools import set_sum
interior_deps = set_sum(iflux.dependencies
for iflux in interiors)
boundary_int_deps = set_sum(bflux.int_dependencies
for bflux in boundaries)
boundary_ext_deps = set_sum(bflux.ext_dependencies
for bflux in boundaries)
self.interior_deps = list(interior_deps)
self.boundary_int_deps = list(boundary_int_deps)
self.boundary_ext_deps = list(boundary_ext_deps)
self.boundary_deps = list(boundary_int_deps | boundary_ext_deps)
self.dep_to_tag = {}
for bflux in boundaries:
for dep in get_flux_dependencies(
bflux.flux_expr, bflux.bpair, bdry="ext"):
self.dep_to_tag[dep] = bflux.bpair.tag
def __init__(self, is_lift, interiors, boundaries, quadrature_tag):
from hedge.optemplate.tools import get_flux_dependencies
self.is_lift = is_lift
self.interiors = tuple(interiors)
self.boundaries = tuple(boundaries)
self.quadrature_tag = quadrature_tag
from pytools import set_sum
interior_deps = set_sum(iflux.dependencies for iflux in interiors)
boundary_int_deps = set_sum(bflux.int_dependencies
for bflux in boundaries)
boundary_ext_deps = set_sum(bflux.ext_dependencies
for bflux in boundaries)
self.interior_deps = list(interior_deps)
self.boundary_int_deps = list(boundary_int_deps)
self.boundary_ext_deps = list(boundary_ext_deps)
self.boundary_deps = list(boundary_int_deps | boundary_ext_deps)
self.dep_to_tag = {}
for bflux in boundaries:
for dep in get_flux_dependencies(bflux.flux_expr,
bflux.bpair,
bdry="ext"):
self.dep_to_tag[dep] = bflux.bpair.tag
def write_boundary_flux_code(self, for_benchmark):
given = self.plan.given
flux_write_code = Block()
fluxes_by_bdry_number = {}
for flux_nr, wdflux in enumerate(self.fluxes):
for bflux_info in wdflux.boundaries:
if for_benchmark:
bdry_number = 0
else:
bdry_number = self.executor.boundary_tag_to_number[
bflux_info.bpair.tag]
fluxes_by_bdry_number.setdefault(bdry_number, [])\
.append((flux_nr, bflux_info))
flux_write_code.extend([
Initializer(
MaybeUnused(POD(given.float_type, "flux%d" % flux_nr)),
0)
for flux_nr in range(len(self.fluxes))])
for bdry_number, nrs_and_fluxes in fluxes_by_bdry_number.iteritems():
bblock = []
from pytools import set_sum
int_deps = set_sum(flux_rec.int_dependencies
for flux_nr, flux_rec in nrs_and_fluxes)
ext_deps = set_sum(flux_rec.ext_dependencies
for flux_nr, flux_rec in nrs_and_fluxes)
for dep in int_deps:
bblock.extend([
Comment(str(dep)),
Initializer(
MaybeUnused(POD(given.float_type, "val_a_field%d"
% self.dep_to_index[dep])),
"fp_tex1Dfetch(field%d_tex, a_index)" % self.dep_to_index[dep])
])
for dep in ext_deps:
bblock.extend([
Comment(str(dep)),
Initializer(
MaybeUnused(POD(given.float_type, "val_b_field%d"
% self.dep_to_index[dep])),
"fp_tex1Dfetch(field%s_tex, b_index)" % self.dep_to_index[dep])
])
f2cm = FluxToCodeMapper(given.float_type)
comp_code = [Line()]
for flux_nr, flux_rec in nrs_and_fluxes:
comp_code.append(
Statement(("flux%d += " % flux_nr) +
flux_to_code(f2cm, is_flipped=False,
int_field_expr=flux_rec.bpair.field,
ext_field_expr=flux_rec.bpair.bfield,
dep_to_index=self.dep_to_index,
flux=flux_rec.flux_expr, prec=PREC_NONE)))
if f2cm.cse_name_list:
bblock.append(Line())
bblock.extend(
Initializer(
Value("value_type", cse_name), cse_str)
for cse_name, cse_str in f2cm.cse_name_list)
flux_write_code.extend([
Line(),
Comment(nrs_and_fluxes[0][1].bpair.tag),
If("(fpair->boundary_bitmap) & (1 << %d)" % (bdry_number),
Block(bblock+comp_code)),
])
flux_write_code.extend([Line(),]
+[
self.gen_store(flux_nr, "fpair->a_dest+FACEDOF_NR",
"fpair->face_jacobian * flux%d" % flux_nr)
for flux_nr in range(len(self.fluxes))
]
#Assign("debugbuf[blockIdx.x*96+fpair_nr]", "10000+fpair->a_dest"),
)
return flux_write_code
def write_boundary_flux_code(self, for_benchmark):
given = self.plan.given
flux_write_code = Block()
fluxes_by_bdry_number = {}
for flux_nr, wdflux in enumerate(self.fluxes):
for bflux_info in wdflux.boundaries:
if for_benchmark:
bdry_number = 0
else:
bdry_number = self.executor.boundary_tag_to_number[
bflux_info.bpair.tag]
fluxes_by_bdry_number.setdefault(bdry_number, [])\
.append((flux_nr, bflux_info))
flux_write_code.extend([
Initializer(MaybeUnused(POD(given.float_type, "flux%d" % flux_nr)),
0) for flux_nr in range(len(self.fluxes))
])
for bdry_number, nrs_and_fluxes in fluxes_by_bdry_number.iteritems():
bblock = []
from pytools import set_sum
int_deps = set_sum(flux_rec.int_dependencies
for flux_nr, flux_rec in nrs_and_fluxes)
ext_deps = set_sum(flux_rec.ext_dependencies
for flux_nr, flux_rec in nrs_and_fluxes)
for dep in int_deps:
bblock.extend([
Comment(str(dep)),
Initializer(
MaybeUnused(
POD(given.float_type,
"val_a_field%d" % self.dep_to_index[dep])),
"fp_tex1Dfetch(field%d_tex, a_index)" %
self.dep_to_index[dep])
])
for dep in ext_deps:
bblock.extend([
Comment(str(dep)),
Initializer(
MaybeUnused(
POD(given.float_type,
"val_b_field%d" % self.dep_to_index[dep])),
"fp_tex1Dfetch(field%s_tex, b_index)" %
self.dep_to_index[dep])
])
f2cm = FluxToCodeMapper(given.float_type)
comp_code = [Line()]
for flux_nr, flux_rec in nrs_and_fluxes:
comp_code.append(
Statement(
("flux%d += " % flux_nr) +
flux_to_code(f2cm,
is_flipped=False,
int_field_expr=flux_rec.bpair.field,
ext_field_expr=flux_rec.bpair.bfield,
dep_to_index=self.dep_to_index,
flux=flux_rec.flux_expr,
prec=PREC_NONE)))
if f2cm.cse_name_list:
bblock.append(Line())
bblock.extend(
Initializer(Value("value_type", cse_name), cse_str)
for cse_name, cse_str in f2cm.cse_name_list)
flux_write_code.extend([
Line(),
Comment(nrs_and_fluxes[0][1].bpair.tag),
If("(fpair->boundary_bitmap) & (1 << %d)" % (bdry_number),
Block(bblock + comp_code)),
])
flux_write_code.extend(
[
Line(),
] + [
self.gen_store(flux_nr, "fpair->a_dest+FACEDOF_NR",
"fpair->face_jacobian * flux%d" % flux_nr)
for flux_nr in range(len(self.fluxes))
]
#Assign("debugbuf[blockIdx.x*96+fpair_nr]", "10000+fpair->a_dest"),
)
return flux_write_code