Package hedge :: Package backends :: Package cuda :: Module execute :: Class OperatorCompiler
[hide private]
[frames] | no frames]

Class OperatorCompiler

source code


Nested Classes [hide private]

Inherited from compiler.OperatorCompilerBase: FluxBatch, FluxRecord

Instance Methods [hide private]
 
get_contained_fluxes(self, expr)
Recursively enumerate all flux expressions in the expression tree `expr`.
source code
 
internal_map_flux(self, wdflux) source code
 
map_whole_domain_flux(self, wdflux) source code
 
make_flux_batch_assign(self, names, fluxes, kind) source code
 
finalize_multi_assign(self, names, exprs, do_not_return, priority) source code

Inherited from compiler.OperatorCompilerBase: __call__, __init__, aggregate_assignments, assign_to_new_var, collect_diff_ops, collect_flux_exchange_ops, dep_mapper_factory, get_var_name, make_assign, map_common_subexpression, map_diff_op_binding, map_flux_exchange_op_binding, map_mass_op_binding, map_operator_binding, map_planned_flux

Inherited from optemplate.IdentityMapperMixin: map_boundarize, map_boundary_pair, map_diff_base, map_elementwise_max, map_flux_base, map_flux_exchange, map_mass_base, map_normal_component, map_scalar_parameter

Inherited from optemplate.LocalOpReducerMixin: map_diff, map_inverse_mass, map_mass, map_minv_st, map_stiffness, map_stiffness_t

Inherited from optemplate.FluxOpReducerMixin: map_flux, map_lift

Inherited from pymbolic.mapper.IdentityMapperBase: map_call, map_constant, map_function_symbol, map_list, map_lookup, map_negation, map_numpy_array, map_polynomial, map_power, map_product, map_quotient, map_subscript, map_sum, map_variable, map_vector

Inherited from pymbolic.mapper.RecursiveMapper: rec

Inherited from pymbolic.mapper.Mapper: handle_unsupported_expression, map_foreign, map_rational

Inherited from object: __delattr__, __getattribute__, __hash__, __new__, __reduce__, __reduce_ex__, __repr__, __setattr__, __str__

Properties [hide private]

Inherited from object: __class__

Method Details [hide private]

get_contained_fluxes(self, expr)

source code 

Recursively enumerate all flux expressions in the expression tree `expr`. The returned list consists of `ExecutionPlanner.FluxRecord` instances with fields `flux_expr` and `dependencies`.

Overrides: compiler.OperatorCompilerBase.get_contained_fluxes
(inherited documentation)

make_flux_batch_assign(self, names, fluxes, kind)

source code 
Overrides: compiler.OperatorCompilerBase.make_flux_batch_assign