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

Fix parsing instances for pt inductor #1796

Open
wants to merge 1 commit into
base: develop
Choose a base branch
from

Conversation

tenpercent
Copy link
Contributor

@tenpercent tenpercent commented Jan 6, 2025

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 crash

Added 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.

  • I have added tests relevant to the introduced functionality, and the unit tests are passing locally
  • I have added inline documentation which enables the maintainers with understanding the motivation
  • I have removed the stale documentation which is no longer relevant after this pull request
  • (If this change is user-facing) I have added release notes which provide the end users with a brief summary of the improvement from this pull request
  • I have run clang-format on all changed files
  • Any dependent changes have been merged

add unit test for gen instances for gemms

add unit tests for conv and batched gemms

add unit test for preselected gemm instances

apply ruff lint

add license header for the unit test

add inductor pytest to CI

verbose pip install

switch the directory before installing python packages

move the inductor codegen test

try yet another workdir

Update Jenkinsfile

The directory looks right, fixing pip module not found by invoking pip directly

Update Jenkinsfile

invoke pytest directly since the module is not found

Update Dockerfile

Install setuptools

update package structure

bump setuptools

maybe fix data path for library sources

fix library search path for conv instances

fix path in pyproject definition

compare path used in gen_instances with one in pyproject.toml; fix the difference
@tenpercent tenpercent force-pushed the fix-inductor-instance-gen branch from 69f22b3 to 314d924 Compare January 11, 2025 01:28
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