mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-12 01:10:17 +00:00
1ca05a105a
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
29 lines
753 B
Bash
Executable File
29 lines
753 B
Bash
Executable File
#!/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
|