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

[WIP] Draft autoscheduler #831

Draft
wants to merge 1 commit into
base: main
Choose a base branch
from
Draft

[WIP] Draft autoscheduler #831

wants to merge 1 commit into from

Conversation

mtsokol
Copy link
Collaborator

@mtsokol mtsokol commented Jan 7, 2025

WIP

@mtsokol mtsokol self-assigned this Jan 7, 2025
Copy link

codspeed-hq bot commented Jan 7, 2025

CodSpeed Performance Report

Merging #831 will degrade performances by 30.19%

Comparing ms/draft-autoscheduler (219b11f) with main (ada7871)

Summary

❌ 1 regressions
✅ 339 untouched benchmarks

⚠️ Please fix the performance issues or acknowledge them on CodSpeed.

Benchmarks breakdown

Benchmark main ms/draft-autoscheduler Change
test_index_slice[side=100-rank=2-format='gcxs'] 2.4 ms 3.4 ms -30.19%

@mtsokol mtsokol force-pushed the ms/draft-autoscheduler branch from 3e5894a to 219b11f Compare January 8, 2025 12:12
@hameerabbasi
Copy link
Collaborator

I feel we're implementing a lot of IR pass infrastructure which we don't really need -- IMHO it'd be better to define an IR and do it via egglog or MLIR.

@mtsokol
Copy link
Collaborator Author

mtsokol commented Jan 9, 2025

I think the only part of pass infrastructure rewritten here is rewrite_tools.py taken from Julia Finch. The rest is scheduler logic (passes, IR).
I used egglog for the first try but using pure Python and its structural pattern matching ended up more convenient for me (I could directly map Julia implementation, I can have side-effects inside matches).

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.

2 participants