def METHOD_NAME(device):
    capacity = 10
    hashset = o3c.HashSet(capacity, o3c.int64, (3), device)
    keys = o3c.Tensor([[1, 2, 3], [2, 3, 4], [3, 4, 5]],
                      dtype=o3c.int64,
                      device=device)
    buf_indices, masks = hashset.insert(keys)
    assert masks.to(o3c.int64).sum() == 3
    keys = o3c.Tensor([[1, 2, 3], [3, 4, 5], [4, 5, 6]],
                      dtype=o3c.int64,
                      device=device)
    buf_indices, masks = hashset.find(keys)
    np.testing.assert_equal(masks.cpu().numpy().flatten(),
                            np.array([True, True, False]))
    keys = o3c.Tensor([[1, 2, 3], [4, 5, 6]], dtype=o3c.int64, device=device)
    masks = hashset.erase(keys)
    np.testing.assert_equal(masks.cpu().numpy().flatten(),
                            np.array([True, False]))