def METHOD_NAME(self, device, dtype, bs):
    run_comp_nocomp(
        torch_bmm,
        init_tensor([[[1], [2], [3], [4]]] * bs, dtype=dtype, device=device),
        init_tensor([[[1, 2, 3, 4]]] * bs, dtype=dtype, device=device),
    )
    run_comp_nocomp(
        torch_bmm,
        init_tensor([[[1, 2, 3, 4]]] * bs, dtype=dtype, device=device),
        init_tensor([[[1], [2], [3], [4]]] * bs, dtype=dtype, device=device),
    )