kaushikcfd/feinsum

Check dtypes in `fnsm.match_t_unit_to_einsum`

Closed this issue · 1 comments

Check dtypes in `fnsm.match_t_unit_to_einsum`

Done, see

def _get_dtype_for_subst_argument(t_unit: lp.TranslationUnit,
kernel_name: str,
rule_name: str) -> np.dtype[Any]:
"""
Returns the inferred data type for *rule_name*'s expression when all the
arguments passed to it are equal to it kernel's
:attr:`loopy.LoopKernel.index_dtype`.
"""