PanZezhong1725 / operators

算子库
7 stars 2 forks source link

device id问题 #63

Open xgqdut2016 opened 1 month ago

xgqdut2016 commented 1 month ago

dev分支的swiglu,RoPE这两个算子的CUDA重构都没有写 if(cudaSetDevice(desc->device_id) != cudaSuccess){ return STATUS_BAD_DEVICE; }