Fix parsing instances for pt inductor #1796
Open
+66
−9
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Proposed changes
Bug: freshly added I4 instances are parsed incorrectly
Fix: ignore instances which can't be parsed
Testing
With pytorch:
pytest test/inductor/test_ck_backend.py
Standalone:
python -m ck4inductor.universal_gemm.gen_instances
doesn't crashAdded unit test:
pytest test/test_gen_instances.py --log-cli-level=DEBUG
Results: https://gist.github.com/tenpercent/ac158212aaddd38fd211eb4fa470c636
Checklist
Please put an
x
into the boxes that apply. You can also fill these out after creating the PR. If you're not sure, please don't hesitate to ask.clang-format
on all changed files