fix: explicitly set device to CPU for RNG state tensor#2344

yzh119 merged 2 commits intomainflashinfer-ai/flashinfer:mainfrom

claude/issue-2333-20260112-0956flashinfer-ai/flashinfer:claude/issue-2333-20260112-0956Copy head branch name to clipboard