composable_kernel/python/ck4inductor/util.py at 29e58d5b28a7f8490ced9b25c17519d110f7bba7 - composable_kernel - Public git mirror

ROCm/composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 14:29:05 +00:00

Files

Max Podkorytov 29e58d5b28 Make the library which generates CK instances for pytorch2 inductor's CK backend usage

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)

2024-05-22 13:44:22 -07:00

8 lines

138 B

Python

Raw Blame History

 import functools
 import os
 @functools.lru_cache(None)
 def library_path():
     return os.path.join(os.path.dirname(__file__), 'library')