CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
79 lines
2.5 KiB
Markdown
79 lines
2.5 KiB
Markdown
# Composable Kernel Tools
|
|
|
|
This directory contains utility tools for building, testing, and analyzing Composable Kernel.
|
|
|
|
These tools are designed to be LLM-agnostic and can be used with any AI assistant or directly from the command line.
|
|
|
|
## Available Tools
|
|
|
|
### ck-docker
|
|
|
|
Build and test composable_kernel in Docker with ROCm support.
|
|
|
|
See [README_ck-docker.md](README_ck-docker.md) for details.
|
|
|
|
**Quick start:**
|
|
```bash
|
|
# Add to PATH
|
|
export PATH="$PATH:$PWD/script/tools"
|
|
|
|
# Start container and build
|
|
ck-docker start
|
|
ck-docker build test_amdgcn_mma
|
|
ck-docker test test_amdgcn_mma
|
|
```
|
|
|
|
### ck-build-analysis
|
|
|
|
Analyze Composable Kernel build times using Clang's -ftime-trace profiler.
|
|
|
|
See [README_ck-build-analysis.md](README_ck-build-analysis.md) for details.
|
|
|
|
**Quick start:**
|
|
```bash
|
|
# Add to PATH
|
|
export PATH="$PATH:$PWD/script/tools"
|
|
|
|
# Analyze build time
|
|
ck-build-analysis example_convnd_fwd_xdl_fp8
|
|
```
|
|
|
|
## LLM Assistant Integration
|
|
|
|
These tools can be used as-is with any LLM assistant by providing the tool documentation to the assistant. The assistant can then invoke these tools on your behalf.
|
|
|
|
For example, you can ask:
|
|
- "Start the docker container"
|
|
- "Build and test test_amdgcn_mma"
|
|
- "Analyze build time for example_convnd_fwd_xdl_fp8"
|
|
|
|
The assistant will translate your natural language request into the appropriate tool invocation.
|
|
|
|
## Dependencies
|
|
|
|
- **ck-docker**: Requires Docker and ROCm-capable GPU (for running tests)
|
|
- **ck-build-analysis**: Requires Docker, automatically installs Python dependencies (jinja2) via `uv`
|
|
|
|
## Directory Structure
|
|
|
|
```
|
|
script/tools/
|
|
├── README.md # This file
|
|
├── README_ck-docker.md # Documentation for ck-docker
|
|
├── README_ck-build-analysis.md # Documentation for ck-build-analysis
|
|
├── ck-docker # Docker container management tool
|
|
├── ck-build-analysis # Build time analysis tool
|
|
├── common.sh # Shared utilities for bash scripts
|
|
├── analyze_build_trace.py # Python script for trace analysis (PEP 723 compliant)
|
|
└── templates/
|
|
└── build_analysis_report.md.jinja # Jinja2 template for analysis reports
|
|
```
|
|
|
|
## Contributing
|
|
|
|
When adding new tools to this directory:
|
|
1. Keep them LLM-agnostic (avoid hardcoding references to specific AI assistants)
|
|
2. Provide clear command-line usage documentation
|
|
3. Include examples for both CLI and LLM assistant usage
|
|
4. Follow the existing naming convention and structure
|