diff --git a/tests/kernels/attention/test_encoder_decoder_attn.py b/tests/kernels/attention/test_encoder_decoder_attn.py index c6ce7b0cc..8efc701f3 100644 --- a/tests/kernels/attention/test_encoder_decoder_attn.py +++ b/tests/kernels/attention/test_encoder_decoder_attn.py @@ -99,7 +99,7 @@ class TestResources(NamedTuple): Attributes: * scale: 1/sqrt(d) scale factor for attn - * attn_backend: implementatino of abstraction + * attn_backend: implementations of abstraction attention interface using a particular kernel library i.e. XFormers diff --git a/tests/v1/sample/test_topk_topp_sampler.py b/tests/v1/sample/test_topk_topp_sampler.py index 63fdeb5a6..9d695cd91 100644 --- a/tests/v1/sample/test_topk_topp_sampler.py +++ b/tests/v1/sample/test_topk_topp_sampler.py @@ -28,7 +28,7 @@ def reset_default_device(): torch.set_default_device(original_device) -def test_topk_impl_equivalance(): +def test_topk_impl_equivalence(): torch.set_default_device(DEVICE) generator = Generator(device=DEVICE).manual_seed(33)