mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-14 10:09:41 +00:00
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)
[ROCm/composable_kernel commit: 29e58d5b28]
8 lines
138 B
Python
8 lines
138 B
Python
import functools
|
|
import os
|
|
|
|
|
|
@functools.lru_cache(None)
|
|
def library_path():
|
|
return os.path.join(os.path.dirname(__file__), 'library')
|