pub fn if_else_expand(
context: &mut CubeContext,
runtime_cond: ExpandElement,
then_block: impl FnOnce(&mut CubeContext),
) -> IfElseExpand
pub fn if_else_expand(
context: &mut CubeContext,
runtime_cond: ExpandElement,
then_block: impl FnOnce(&mut CubeContext),
) -> IfElseExpand