[drbh]

this pr adds a rocm build for causal-conv1d

• build.toml: add the rocm backend, rocm = "2.10" stable abi and a cpp-rocm kernel with the same sources (same rocm-archs as activation)
• causal_conv1d.cpp includes the hip runtime and aliases cudaStream_t under USE_ROCM since kernel-builder only hipifies the .cu files
• torch_binding.cpp also registers the ops for ROCM_KERNEL. without it the kernel loads but every op raises NotImplementedError

the kernels already have the upstream USE_ROCM paths so no kernel source changes were needed

built all rocm variants (rocm70, rocm71, rocm714, rocm72, rocm100) with nix build .#backendBundle.rocm and ran the tests on mi300x (gfx942) against the bundle with torch 2.10.0+rocm7.0, 2.12.1+rocm7.1 and 2.14.1+rocm7.2

pytest -q tests/test_causal_conv1d_ci.py tests/test_causal_conv1d.py

8802 passed, 3888 skipped

same result on all three. the skips are the existing Only channel_last support initial_states skip. kernels benchmark shows 2.81x / 3.00x / 4.50x over the torch reference for base / large / xlarge

note: #1308 makes the same causal_conv1d.cpp change in the mamba-ssm copy and #1300 touches the same files, so one of these will need a rebase. transformers also needs rocm entries for causal_conv1d_fn and causal_conv1d_update in hub_kernels.py before it picks this up

*keeping as draft until ci support rocm

• feat: run rocm kernel tests on the amd mi300 runners #1313

[github-actions]

This PR changes source code for the following kernel(s): causal-conv1d.

If you have the kernel-bot build permissions, then trigger /kernel-bot build causal-conv1d yourself. Otherwise, a maintainer will review the PR and will take an appropriate action. Thanks for your patience.

[danieldk]

Nice!!!

[drbh]

waiting on #1328 to run build

[drbh]

/kernel-bot build causal-conv1d

[github-actions]

Build request processed.

Command: /kernel-bot build causal-conv1d
Mode: build only
Target branch: pr-1319
PR head SHA: 8f818fba681eeb8cd72d79963a351722af65b5f8
Workflows: build.yaml, build-mac.yaml, build-windows.yaml

Dispatched (1):

• causal-conv1d (build.yaml): https://github.com/huggingface/kernels-community/actions/runs/43476215121