AI/rocm-systems
2
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

msccl: print stack and memory usage (#723)

Browse Source 
* msccl: print stack and memory usage

* Update number of kernels calculation
This commit is contained in:
Wenkai Du authored and GitHub committed 2023-04-14 14:59:03 -07:00
1 parent 53c1c38f0e
commit 4b09ffba43
3 files changed
+21 -1
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/include/msccl/msccl_lifecycle.h
+2
View File
@@ -32,4 +32,6 @@ ncclResult_t mscclGroupEnd();
ncclResult_t mscclTeardown();
size_t mscclKernMaxLocalSize();
#endif