mirage-project / mirage

A multi-level tensor algebra superoptimizer
Apache License 2.0
262 stars 16 forks source link

Wrap up device memory manager to avoid unnecessary memory allocation #19

Closed wmdi closed 3 weeks ago

wmdi commented 1 month ago

This PR closes #5. It implements a wrapper upon the device memory manager that only calculates the memory consumption when a tensor is created and allocates the actual physical memory only when necessary (i.e., profiling and fingerprint checking).

It also helps reduce side effects of graph operator generation, which is important for parallelizing the search procedure.

TODO: