[Bitmask] Add allocate_token_bitmask to xgr from testing

[CharlieFRuan]

This PR changes from from xgrammar.testing import _allocate_token_bitmask to from xgrammar import allocate_token_bitmask.

Passed all test with pytest .