Fixes a typo and adds missing constants matching cuda Change-Id: Iebf47d107af361cbd356c2a3b9cd6ef121d01363