fix: fix device of temp_mask when using cuda#11

Hniii98 wants to merge 1 commit intoInfiniTensor:mainInfiniTensor/llaisys:mainfrom

Hniii98:fix/test/ops/self_attention.pyHniii98/llaisys:fix/test/ops/self_attention.pyCopy head branch name to clipboard