Add device guards in pytorch cuda kernels.#60

hzhangxyz merged 1 commit into

mainUSTC-KnowledgeComputingLab/qmp-kit:mainfrom

fix/mismatch-cuda-device-and-streamUSTC-KnowledgeComputingLab/qmp-kit:fix/mismatch-cuda-device-and-streamCopy head branch name to clipboard