Function if_else_expr_expand

pub fn if_else_expr_expand<C: CubePrimitive>(
    context: &mut CubeContext,
    runtime_cond: ExpandElement,
    then_block: impl FnOnce(&mut CubeContext) -> ExpandElementTyped<C>,
) -> IfElseExprExpand<C>