dcegm.interfaces.jit_large_arrays

Functions

split_structure_and_batch_info(model_structure, batch_info)

Splits the model structure and batch info into static parts, which we can not jit

merge_non_jit_and_jit_model_structure(...)

Generate one model_structure to handle inside the package functions.

merg_non_jit_batch_info_and_jit_batch_info(...)

Module Contents

dcegm.interfaces.jit_large_arrays.split_structure_and_batch_info(model_structure, batch_info)

Splits the model structure and batch info into static parts, which we can not jit compile and (large) arrays that we want to include in the function call for jitting.

dcegm.interfaces.jit_large_arrays.merge_non_jit_and_jit_model_structure(model_structure_jit, model_structure_non_jit)

Generate one model_structure to handle inside the package functions.

dcegm.interfaces.jit_large_arrays.merg_non_jit_batch_info_and_jit_batch_info(batch_info_jit, batch_info_non_jit)