Skip to content

Commit 3a4c3a1

Browse files
authored
Fix ExecutorTorch → ExecuTorch in comments only
Differential Revision: D104074211 Pull Request resolved: #19337
1 parent cdcc915 commit 3a4c3a1

3 files changed

Lines changed: 4 additions & 4 deletions

File tree

backends/mlx/llm/cache.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323

2424
class KVCache(nn.Module):
2525
"""
26-
MLX-optimized KV cache with ExecutorTorch llama KVCache interface.
26+
MLX-optimized KV cache with ExecuTorch llama KVCache interface.
2727
2828
This class follows the same interface as examples/models/llama/attention.py KVCache,
2929
making it a drop-in replacement, but uses the mlx::kv_cache_update op internally

backends/mlx/llm/et_attention.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
# LICENSE file in the root directory of this source tree.
88

99
"""
10-
MLX-optimized attention for ExecutorTorch's Llama attention registry.
10+
MLX-optimized attention for ExecuTorch's Llama attention registry.
1111
1212
Registers an "mlx" attention type that uses mlx::kv_cache_update and
1313
mlx::custom_sdpa for efficient execution on Apple Silicon.

backends/mlx/test/test_ops.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1810,7 +1810,7 @@ class KVCacheModel(nn.Module):
18101810
"""
18111811
Test model wrapping KVCache from cache.py.
18121812
1813-
This tests the ExecutorTorch llama KVCache-compatible interface that uses
1813+
This tests the ExecuTorch llama KVCache-compatible interface that uses
18141814
the mlx::kv_cache_update op internally.
18151815
"""
18161816

@@ -1845,7 +1845,7 @@ def forward(
18451845
@register_test
18461846
class KVCacheTest(OpTestCase):
18471847
"""
1848-
Test case for MLX KVCache with ExecutorTorch llama KVCache interface.
1848+
Test case for MLX KVCache with ExecuTorch llama KVCache interface.
18491849
18501850
This verifies that KVCache:
18511851
1. Accepts the ET llama KVCache update interface

0 commit comments

Comments
 (0)