Skip to content
Open
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion src/enqueue.cc
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,10 @@ ncclResult_t ncclInitKernelsForDevice(int cudaArch, int maxSharedMem, size_t* ma
if (fn == nullptr) continue;

cudaError_t errcode = cudaFuncGetAttributes(&attr, fn);
if (errcode != cudaSuccess) continue; // Silently ignore failures
if (errcode != cudaSuccess) {
errorcode = cudaGetLastError(); // Drain error code
continue; // Silently ignore failures
}
if (maxStackSize) {
if (attr.localSizeBytes > *maxStackSize) *maxStackSize = attr.localSizeBytes;
}
Expand Down