def METHOD_NAME(attrs, inputs, out_type, target):
    """Batch matmul strategy for ROCM"""
    strategy = batch_matmul_strategy_cuda(attrs, inputs, out_type, target)
    if target.kind.name == "rocm" and "rocblas" in target.libs:
        assert out_type.dtype == inputs[0].dtype, "Mixed precision not supported."
        strategy.add_implementation(
            wrap_compute_batch_matmul(topi.rocm.batch_matmul_rocblas),
            wrap_topi_schedule(topi.rocm.schedule_batch_matmul_rocblas),
            name="batch_matmul_rocblas.rocm",
            plevel=12,
        )
    return strategy