Commit cfe3238
dyzheng
Feat(psi): add PsiStorageMode and device_memory_mode for GPU k-point paging
- Add PsiStorageMode enum (ALL_GPU, ALL_CPU, PAGED_GPU) to psi namespace
- Add paging interface: load_k_to_gpu, store_k_from_gpu, ensure_k_on_gpu
- Modify resize() to allocate CPU storage + single-k GPU buffer in PAGED_GPU mode
- Create psi_paging.cpp with paging method implementations
- Add device_memory_mode INPUT parameter (''/full_gpu/paged)
- Add device_memory_mode reading in read_input_item_system.cpp1 parent bb2cb4b commit cfe3238
6 files changed
Lines changed: 567 additions & 47 deletions
File tree
- source
- source_io/module_parameter
- source_psi
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
70 | 70 | | |
71 | 71 | | |
72 | 72 | | |
| 73 | + | |
73 | 74 | | |
74 | 75 | | |
75 | 76 | | |
| |||
Lines changed: 22 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
743 | 743 | | |
744 | 744 | | |
745 | 745 | | |
| 746 | + | |
| 747 | + | |
| 748 | + | |
| 749 | + | |
| 750 | + | |
| 751 | + | |
| 752 | + | |
| 753 | + | |
| 754 | + | |
| 755 | + | |
| 756 | + | |
| 757 | + | |
| 758 | + | |
| 759 | + | |
| 760 | + | |
| 761 | + | |
| 762 | + | |
| 763 | + | |
| 764 | + | |
| 765 | + | |
| 766 | + | |
| 767 | + | |
746 | 768 | | |
747 | 769 | | |
748 | 770 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
2 | 2 | | |
3 | 3 | | |
4 | 4 | | |
| 5 | + | |
5 | 6 | | |
6 | 7 | | |
7 | 8 | | |
| |||
0 commit comments