Skip to content

Commit

Permalink
Revert back to int64_t.
Browse files Browse the repository at this point in the history
  • Loading branch information
dranger003 committed Apr 28, 2024
1 parent 91c10ef commit 0258f9b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ggml-cuda.cu
Original file line number Diff line number Diff line change
Expand Up @@ -1225,7 +1225,7 @@ static void ggml_cuda_op_mul_mat_cublas(

// the main device has a larger memory buffer to hold the results from all GPUs
// ldc == nrows of the matrix that cuBLAS writes into
int ldc = id == ctx.device ? ne0 : row_diff;
int64_t ldc = id == ctx.device ? ne0 : row_diff;

const int compute_capability = ggml_cuda_info().devices[id].cc;

Expand Down

0 comments on commit 0258f9b

Please sign in to comment.