INNER CODE UNIT · Python

get_memory_pool

sangmichaelxie/doremi · flash-attention/csrc/cutlass/tools/library/scripts/pycutlass/src/pycutlass/__init__.py:50

def get_memory_pool(init_pool_size=0, max_pool_size=2**34):
    this.memory_pool = PoolMemoryManager(
        init_pool_size=init_pool_size,
        max_pool_size=max_pool_size
    )
    return this.memory_pool

View source record →