Macro cubecl_core::cpa

macro_rules! cpa {
    ($scope:expr, $out:ident = $lhs:ident + $rhs:expr) => { ... };
    ($scope:expr, $out:ident += $input:ident) => { ... };
    ($scope:expr, $out:ident = add($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = $lhs:ident - $rhs:expr) => { ... };
    ($scope:expr, $out:ident = sub($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = $lhs:ident * $rhs:expr) => { ... };
    ($scope:expr, $out:ident *= $input:ident) => { ... };
    ($scope:expr, $out:ident = mul($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = $lhs:ident / $rhs:expr) => { ... };
    ($scope:expr, $out:ident = div($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = $lhs:ident % $rhs:expr) => { ... };
    ($scope:expr, $out:ident = modulo($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = powf($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = $lhs:ident && $rhs:expr) => { ... };
    ($scope:expr, $out:ident = and($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = $lhs:ident || $rhs:expr) => { ... };
    ($scope:expr, $out:ident = or($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = !$input:expr) => { ... };
    ($scope:expr, $out:ident = not($input:expr)) => { ... };
    ($scope:expr, $out: ident = $lhs:ident & $rhs:ident) => { ... };
    ($scope:expr, $out:ident = bitwise_and($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out: ident = $lhs:ident ^ $rhs:ident) => { ... };
    ($scope:expr, $out:ident = bitwise_xor($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out: ident = $lhs:ident << $rhs:ident) => { ... };
    ($scope:expr, $out:ident = shift_left($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out: ident = $lhs:ident >> $rhs:ident) => { ... };
    ($scope:expr, $out:ident = shift_right($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = $lhs:ident == $rhs:expr) => { ... };
    ($scope:expr, $out:ident = equal($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = $lhs:ident != $rhs:expr) => { ... };
    ($scope:expr, $out:ident = not_equal($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = $lhs:ident > $rhs:expr) => { ... };
    ($scope:expr, $out:ident = greater($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = $lhs:ident >= $rhs:expr) => { ... };
    ($scope:expr, $out:ident = greater_equal($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = $lhs:ident < $rhs:expr) => { ... };
    ($scope:expr, $out:ident = lower($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = $lhs:ident <= $rhs:expr) => { ... };
    ($scope:expr, $out:ident = lower_equal($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = max($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = min($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = $lhs:ident[$rhs:expr]) => { ... };
    ($scope:expr, $out:ident = index($lhs:expr, $rhs:expr)) => { ... };
    ($scope:expr, $out:ident = unchecked($lhs:ident[$rhs:expr])) => { ... };
    ($scope:expr, $out:ident[$lhs:ident] = $rhs:expr) => { ... };
    ($scope:expr, unchecked($out:ident[$lhs:ident]) = $rhs:expr) => { ... };
    ($scope:expr, $out:ident = |$input:ident|) => { ... };
    ($scope:expr, $out:ident = abs($input:expr)) => { ... };
    ($scope:expr, $out:ident = exp($input:expr)) => { ... };
    ($scope:expr, $out:ident = log($input:expr)) => { ... };
    ($scope:expr, $out:ident = log1p($input:expr)) => { ... };
    ($scope:expr, $out:ident = cos($input:expr)) => { ... };
    ($scope:expr, $out:ident = sin($input:expr)) => { ... };
    ($scope:expr, $out:ident = tanh($input:expr)) => { ... };
    ($scope:expr, $out:ident = sqrt($input:expr)) => { ... };
    ($scope:expr, $out:ident = floor($input:expr)) => { ... };
    ($scope:expr, $out:ident = ceil($input:expr)) => { ... };
    ($scope:expr, $out:ident = erf($input:expr)) => { ... };
    ($scope:expr, $out:ident = $input:ident) => { ... };
    ($scope:expr, $out:ident = vec4($a:ident,$b:ident,$c:ident,$d:ident)) => { ... };
    ($scope:expr, $out:ident = $input:ident) => { ... };
    ($scope:expr, $out:ident = cast($input:expr)) => { ... };
    ($scope:expr, $out:ident = shape($input:expr, $dim:expr)) => { ... };
    ($scope:expr, $out:ident = stride($input:expr, $dim:expr)) => { ... };
    ($scope:expr, $out:ident = len($input:expr)) => { ... };
    ($scope:expr, range($start:expr, $end:expr).for_each($arg:expr)) => { ... };
    ($scope:expr, range($start:expr, $end:expr, $unroll:expr).for_each($arg:expr)) => { ... };
    ($scope:expr, loop($arg:expr)) => { ... };
    ($scope:expr, if ($cond:expr).then($arg:expr)) => { ... };
    ($scope:expr, if ($cond:expr).then($arg_if:expr).else($arg_else:expr)) => { ... };
    (binary $lhs:expr, $rhs:expr, $out:expr) => { ... };
    (unary $input:expr, $out:expr) => { ... };
}

Cube Pseudo Assembly.