ColossalAI/colossalai/autochunk
oahzxl 1a6d2a740b take apart chunk code gen 2023-01-06 14:14:45 +08:00
..
autochunk_codegen.py take apart chunk code gen 2023-01-06 14:14:45 +08:00
chunk_region_search.py take apart chunk code gen 2023-01-06 14:14:45 +08:00
chunk_selector.py take apart chunk code gen 2023-01-06 14:14:45 +08:00
index_tracer.py take apart chunk code gen 2023-01-06 14:14:45 +08:00
memory_estiamtor.py take apart chunk code gen 2023-01-06 14:14:45 +08:00
utils.py take apart chunk code gen 2023-01-06 14:14:45 +08:00