Add KernelArg impl for u32 and f64.

This adds KernelArg implementations for u32 and f64.
Adding other types could also be useful in the future.