derek.holloway
7502bf64a4
rocisa
A Python/C++ code generator for ROCm ISA, built with nanobind.
Developer Setup
Install rocisa as an editable package using the invoke task from the tensilelite root:
cd rocm-libraries/projects/hipblaslt/tensilelite
invoke rocisa
This compiles the C++ extension and installs it into your active venv so that
import rocisa works from anywhere — no PYTHONPATH required.
Linux only. The
invokedev workflow (invoke rocisa,invoke build-client) is supported on Linux (a ROCm dev container) only: it usesamdclang, defaults to/opt/rocm, and resolves dependencies via RPATH — none of which apply on Windows. On Windows, rocisa is built by the integrated CMake build (TheRock / CI), which is where its Windows dependent-DLL handling lives.
Rebuilding after C++ changes
See tensilelite/README.md — "Rebuilding rocisa after C++ changes" for the full rebuild instructions, including which cmake build directory to use depending on how rocisa was installed.
Importing rocisa with stale bindings raises an ImportError with a clear rebuild
hint, so you will not silently use an out-of-date extension.
Building independently (without tensilelite)
cd rocisa
pip install -e .
scikit-build-core handles the cmake configuration and compilation automatically.
Requires the ROCm SDK (amdclang++) and /opt/rocm on the default search path,
or set ROCM_PATH.
For more information, see docs/.