mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-04-20 06:49:15 +00:00
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)
This commit is contained in:
36
pyproject.toml
Normal file
36
pyproject.toml
Normal file
@@ -0,0 +1,36 @@
|
||||
[build-system]
|
||||
requires = ["setuptools", "setuptools-scm"]
|
||||
build-backend = "setuptools.build_meta"
|
||||
|
||||
[project]
|
||||
name = "rocm-composable-kernel"
|
||||
dynamic = ["version"]
|
||||
description = "Composable Kernel, performance-critical kernels for machine learning workloads"
|
||||
readme = "README.md"
|
||||
requires-python = ">=3.8"
|
||||
license = {file = "LICENSE"}
|
||||
classifiers = [
|
||||
"Programming Language :: Python :: 3",
|
||||
"License :: OSI Approved :: MIT License",
|
||||
"Operating System :: OS Independent",
|
||||
]
|
||||
dependencies = []
|
||||
|
||||
[project.urls]
|
||||
"Homepage" = "https://github.com/rocm/composable_kernel"
|
||||
"Bug Tracker" = "https://github.com/rocm/composable_kernel/issues"
|
||||
|
||||
[tool.setuptools]
|
||||
packages = ["ck4inductor", "ck4inductor.include", "ck4inductor.library"]
|
||||
|
||||
[tool.setuptools.package-dir]
|
||||
ck4inductor = "python/ck4inductor"
|
||||
"ck4inductor.include" = "include"
|
||||
"ck4inductor.library" = "library"
|
||||
|
||||
[tool.setuptools.package-data]
|
||||
"ck4inductor.include" = ["ck/**/*.hpp"]
|
||||
"ck4inductor.library" = ["src/tensor_operation_instance/gpu/gemm_universal/**/*.hpp"]
|
||||
|
||||
[tool.setuptools.dynamic]
|
||||
version = { attr = "setuptools_scm.get_version" }
|
||||
Reference in New Issue
Block a user