def METHOD_NAME(module):
    for child in module.children():
        if isinstance(child, DeepSpeedTransformerInference):
            return child.attention.attn_qkvw.dtype
        else:
            found_dtype = METHOD_NAME(child)
            if found_dtype:
                return found_dtype