Module flash_attention_ck 

Flash Attention CK installer — ports scripts/install_flash_attention_ck.sh.

Constructs correct cmake commands with HIP compiler flags and ROCm path, make commands with job count, and pip install from build output.

Flash Attention CK depends on PyTorch (ROCm-enabled) and ROCm.

Validation Assertions

• VAL-INSTALL-008: Flash Attention correct cmake command
• VAL-INSTALL-009: Flash Attention correct make command
• VAL-INSTALL-014: Source builds parse build output for errors
• VAL-INSTALL-044: Flash Attention declares dependency on PyTorch and ROCm

Structs

FlashAttentionConfig

Configuration for the Flash Attention CK installer.

FlashAttentionInstaller

The Flash Attention CK installer.

ShellCommand

A constructed shell command.

Enums

GpuArch

GPU architecture specification for building.

InstallMethod

Installation method.