Skip to content

GitLab

Unverified Commit 62b7dd01 authored by PanZezhong1725's avatar PanZezhong1725 Committed by GitHub
Browse files

Merge pull request #198 from InfiniTensor/issue/197 

issue/197/fix  修复rope cuda算子中dimx使用uint32_t代替unsigned int
parents b6bc8a23 ad7124c9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/infiniop/ops/rope/cuda/rope_cuda.cu
View file @ 62b7dd01
...@@ -46,8 +46,8 @@ infiniStatus_t calculateRoPE(const RoPEInfo &info, ...@@ -46,8 +46,8 @@ infiniStatus_t calculateRoPE(const RoPEInfo &info,
const Tdata *sin_table, const Tdata *sin_table,
const Tdata *cos_table, const Tdata *cos_table,
cudaStream_t stream) { cudaStream_t stream) {
auto dimx = unsigned int(info.seqlen), auto dimx = uint32_t(info.seqlen),
dimy = unsigned int(info.nhead); dimy = uint32_t(info.nhead);
int nthreads = std::max(int(info.table_dim), block_size); int nthreads = std::max(int(info.table_dim), block_size);
ropeThreadPerItem<<<dim3(dimx, dimy), nthreads, 0, stream>>>( ropeThreadPerItem<<<dim3(dimx, dimy), nthreads, 0, stream>>>(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment