Make the library which generates CK instances for pytorch2 inductor's CK backend usage

Browse Source

Also bundle the CK library and include files with the pip package.

The package is pip-installable with
`pip install
git+https://github.com/tenpercent/composable_kernel@enable-pip`

(substitute the repo path and branch if necessary)

Testing:

`myenv/bin/python3 -m ck4inductor.universal_gemm.gen_instances`

(prints a list of instances)

`tree myenv/lib/python3.12/site-packages/ck4inductor`

(observe the list of sources along the installed package)

...

This commit is contained in:

Max Podkorytov

2024-05-21 20:37:26 +00:00

parent fd72380aeb

commit 29e58d5b28

5 changed files with 708 additions and 0 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

0

python/ck4inductor/__init__.py Normal file