fix test error regarding logits_types#2918

aleozlx merged 2 commits intoflashinfer-ai:mainflashinfer-ai/flashinfer:mainfrom

aleozlx:logits_type_test_erraleozlx/flashinfer:logits_type_test_errCopy head branch name to clipboard