Skip to content

device id问题 #63

@xgqdut2016

Description

@xgqdut2016

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

Metadata

Metadata

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions