mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-12 01:10:17 +00:00
[rocm-libraries] ROCm/rocm-libraries#6434 (commit 87aae5c)
Fix ck4inductor conv instance parsing for NumGroupsToMerge parameter (#6434) ## Summary - Add `num_groups_to_merge` field to `CKGroupedConvFwdOp` dataclass to match the new (#4273) `NumGroupsToMerge` template parameter added to `DeviceGroupedConvFwdMultipleABD_Xdl_CShuffle_V3` - Enable inductor tests by default in Jenkins CI ## Test plan - [x] Built wheel without patch: `test_gen_conv_instances` fails with `TypeError: takes from 47 to 50 positional arguments but 51 were given` - [x] Built wheel with patch: `test_gen_conv_instances` passes
This commit is contained in:
committed by
assistant-librarian[bot]
parent
cbfb3e242e
commit
1ca05a105a
28
script/run_inductor_tests.sh
Executable file
28
script/run_inductor_tests.sh
Executable file
@@ -0,0 +1,28 @@
|
||||
#!/bin/bash
|
||||
# Run inductor codegen tests
|
||||
# This script is called from Jenkinsfile to reduce pipeline bytecode size
|
||||
|
||||
set -e
|
||||
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
CK_DIR="$(dirname "$SCRIPT_DIR")"
|
||||
VENV_DIR="${WORKSPACE:-/tmp}/ck-inductor-venv"
|
||||
export UV_CACHE_DIR="${WORKSPACE:-/tmp}/.uv-cache"
|
||||
|
||||
cd "$CK_DIR"
|
||||
|
||||
echo "Setting up Python virtual environment at $VENV_DIR"
|
||||
python3 -m venv "$VENV_DIR"
|
||||
. "$VENV_DIR/bin/activate"
|
||||
|
||||
echo "Installing uv for faster package installation"
|
||||
pip install uv
|
||||
|
||||
echo "Installing test dependencies"
|
||||
uv pip install pytest build setuptools setuptools_scm
|
||||
|
||||
echo "Installing ck4inductor package"
|
||||
uv pip install .
|
||||
|
||||
echo "Running inductor codegen tests"
|
||||
python3 -m pytest python/test/test_gen_instances.py -v
|
||||
Reference in New Issue
Block a user