Unverified Commit e6f6ce27 authored by Zihao Ye's avatar Zihao Ye Committed by GitHub
Browse files

[hotfix] Fix cuda illegal memory access error in SpMMCoo. (#2015)

* up

* pylint

* upd
parent 4672c022
......@@ -224,7 +224,7 @@ void SpMMCoo(
int64_t out_size = out.NumElements();
const int nt = FindNumThreads(out_size);
const int nb = (out_size + nt - 1) / nt;
_FillKernel<<<nt, nb, 0, thr_entry->stream>>>(out_data, out_size, ReduceOp::zero);
_FillKernel<<<nb, nt, 0, thr_entry->stream>>>(out_data, out_size, ReduceOp::zero);
const int ntx = FindNumThreads(len);
const int nty = CUDA_MAX_NUM_THREADS / ntx;
......
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