pub fn if_expand(
context: &mut CubeContext,
runtime_cond: ExpandElement,
block: impl FnOnce(&mut CubeContext),
)
pub fn if_expand(
context: &mut CubeContext,
runtime_cond: ExpandElement,
block: impl FnOnce(&mut CubeContext),
)