Commit Graph

7 Commits

Author SHA1 Message Date
Johannes Graner
8af66c65d0 Update pre-commit to fixed versions, run remod for ck_tile (#2895)
* Fix ruff linter errors

* Fix remod dos2unix command

* Clang format

* Ignore utility in remod

* Run remod

* Specify clang-format version in pre-commit

* Specify ruff version

* Include PoolKernelArgs in reference_pool

* Add calculate_total_elements to reference batched contraction

* Fix calculate_total_elements declaration

* Refactor remod pre-commit hook

* Fix Aquant tests

---------

Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>

[ROCm/composable_kernel commit: d40b50b9d5]
2025-10-16 15:29:17 -07:00
Max Podkorytov
1eafaa321f tweak version (#2954)
[ROCm/composable_kernel commit: a76c7b1028]
2025-10-01 15:00:41 -07:00
Max Podkorytov
971203e15a fix parsing instances for pt inductor (#1796)
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

Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>

[ROCm/composable_kernel commit: c0b90f130f]
2025-01-13 13:51:08 -08:00
Max Podkorytov
f21ced64aa [Python] Add batched gemm instances parsing (#1684)
* add op

* do not insert ds parameters as they are already parsed

* reset ds parameters

* apply ruff

[ROCm/composable_kernel commit: 44828b7c0f]
2024-11-30 08:11:42 -08:00
Max Podkorytov
d99a3611fc add parsing grouped conv fwd instances
[ROCm/composable_kernel commit: eda5938386]
2024-10-25 08:25:53 -07:00
Max Podkorytov
3aeb97f3ec modify python wrapper for addmm (#1441)
[ROCm/composable_kernel commit: 886d14ccb2]
2024-08-06 15:09:27 -07:00
Max Podkorytov
564de0adc0 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)


[ROCm/composable_kernel commit: 29e58d5b28]
2024-05-22 13:44:22 -07:00