ColossalAI/tests/test_tensor/test_chunk.py at 2b2dc1c86becfa253efbc7bf8a6e8d3dd39b08b8

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-12 10:54:49 +00:00

Files

ver217 be01db37c8 [tensor] refactor chunk mgr and impl MemStatsCollectorV2 (#1077 )

* polish chunk manager

* polish unit test

* impl add_extern_static_tensor for chunk mgr

* add mem stats collector v2

* polish code

* polish unit test

* polish code

* polish get chunks

2022-06-09 20:56:34 +08:00

3.1 KiB

Raw Blame History

View Raw

3.1 KiB Raw Blame History

3.1 KiB

Raw Blame History