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

LoRA manual MLIR model #171

Merged
merged 4 commits into from
Sep 10, 2024
Merged

Conversation

adam-smnk
Copy link
Collaborator

@adam-smnk adam-smnk commented Sep 10, 2024

Adds manually written MLIR LoRA snippet and further parametrizes benchmark model creation.

The MLIR snippet creates weights as constants to allow for compile-time packing. Function arguments and operations are kept in line with subgraph created by OV MLIR outlining.
Benchmark model builders are parametrized to allow control over weights matrix and LoRA dimensions.

The new model's performance is measured as 'manual MLIR' which evaluates pure MLIR performance with more optimal input IR compared to the current OV outlining.

@adam-smnk adam-smnk changed the title LoRA manual MLIRmodel LoRA manual MLIR model Sep 10, 2024
@adam-smnk adam-smnk requested a review from rolfmorel September 10, 2024 12:16
@adam-smnk adam-smnk merged commit 2063c7f into slyalin:mlir Sep 10, 2024
13 of 29 checks passed
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