Fixed some things from my previous commit to the rmm kernel + a few comments

Currently, this rmm kernel runs ~5% faster (compared to master) for
caroteno test on a K20, <1% faster for hemo test (not quite sure why...)