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

benchmark_litgpt.py: fix executors to be a list; fix the order of executors for sdpa #1006

Merged
merged 4 commits into from
Aug 21, 2024

Conversation

IvanYashchuk
Copy link
Collaborator

@IvanYashchuk IvanYashchuk commented Aug 21, 2024

A follow-up fix for #1002. default_executors returns a tuple that can't be modified. I'm sorry I haven't tested the code properly.

In addition, this PR removes "cudnn" reading from the compile option because it's part of the default executors.

cc @crcrpar

@IvanYashchuk IvanYashchuk changed the title benchmark_litgpt.py: Start building executors list from the default ones, second attempt benchmark_litgpt.py: fix executors to be a list; fix the order of executors for sdpa Aug 21, 2024
Copy link
Collaborator

@t-vi t-vi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you @IvanYashchuk

@t-vi t-vi enabled auto-merge (squash) August 21, 2024 11:05
@t-vi t-vi merged commit 133953c into main Aug 21, 2024
37 checks passed
@t-vi t-vi deleted the benchmark-litgpt-default-executors branch August 21, 2024 12:10
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants