[PATCH 5.9 694/757] habanalabs: cast to u64 before shift > 31 bits