msccl: print stack and memory usage (#723)

* msccl: print stack and memory usage * Update number of kernels calculation [ROCm/rccl commit: 4b09ffba43]
2023-04-14 14:59:03 -07:00
parent aeffdf872b
commit 1af52edd6a
3 changed files with 21 additions and 1 deletions
@@ -32,4 +32,6 @@ ncclResult_t mscclGroupEnd();

 ncclResult_t mscclTeardown();

+size_t mscclKernMaxLocalSize();
+
 #endif