def METHOD_NAME(
    self, compute_kernel: EmbeddingComputeKernel, dtype: torch.dtype
) -> None:
    config = self._create_config(compute_kernel)
    sharded = GroupedPooledEmbeddingsLookup(
        grouped_configs=[config],
        device=torch.device("cuda:0"),
    )
    quantized = quantize_sharded_embeddings(sharded, dtype=dtype)
    for _, buffer in quantized.named_buffers():
        self.assertEqual(buffer.dtype, torch.uint8)