msccl: print stack and memory usage (#723)

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

 ncclResult_t mscclTeardown();

+size_t mscclKernMaxLocalSize();
+
 #endif