habanalabs: add RMWREG32_SHIFTED to set a val within a mask