We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 62fa829 commit feaed9dCopy full SHA for feaed9d
megatron/core/inference/unified_memory.py
@@ -42,7 +42,7 @@ def compile_allocator():
42
43
if _compilation_state != CompilationState.UNATTEMPTED:
44
return
45
-
+
46
_mempool_c_src = r"""
47
#include <cuda_runtime_api.h>
48
#include <cstddef>
0 commit comments