Function test_executor_flow

rl_engine/tests/test_dispatch.py:45–49

()

    registry = KernelRegistry()

    assert registry._priority_map["rocm"]["attn"][0] == OpBackend.PYTORCH_ATTN
    assert registry._priority_map["rocm"]["attn"][1] == OpBackend.ROCM_FLASH_ATTN


class TestMusaPlatform:
    @staticmethod
    def _mock_torch_musa_import(monkeypatch):
        monkeypatch.setattr(