Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Refactor: Integrate the local and global memory-planner #193

Merged
merged 28 commits into from
Nov 10, 2024
Merged

Conversation

hikettei
Copy link
Owner

@hikettei hikettei commented Nov 10, 2024

So the bottleneck in group-schedule is the following two stuff:

  • (apply-reduce+move-fusion schedule graph)
  • (apply-serialize-reduction schedule graph)
  • todo: fix memory planner
    • remove memory-planner-global
    • remove gradxxx
    • fix the input guard (make-tensor ... :from 'x) <- this won't be changed to in-place by memory planner

@hikettei hikettei changed the title Optimize JIT Compiler optimize jit (remove id->users) + refactored memory planner Nov 10, 2024
@hikettei hikettei changed the title optimize jit (remove id->users) + refactored memory planner Integrate the local and global memory planner Nov 10, 2024
@hikettei hikettei changed the title Integrate the local and global memory planner Refactor: Integrate the local and global memory-planner Nov 10, 2024
@hikettei hikettei mentioned this pull request Nov 10, 2024
28 tasks
@hikettei hikettei marked this pull request as ready for review November 10, 2024 10:18
@hikettei hikettei merged commit 62a7873 into main Nov 10, 2024
6 checks passed
@hikettei hikettei deleted the fastcompile branch November 10, 2024 10:22
hikettei added a commit that referenced this pull request Nov 10, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant