unaryPow

import numir;
import mir.ndslice;
import grain.kernel;
import mir.math : pow;

auto x = iota([2, 3], 1).as!float
    .slice
    .variable
    .to!DeviceStorage;
unaryPow(x, 2f);
assert(approxEqual(x.to!HostStorage.sliced, iota([2, 3], 1).as!float
        .map!(x => pow(x, 2))));