[XLA:CPU] Fix extension typos

PiperOrigin-RevId: 707491122

Author: WillFroom

xla/backends/cpu/testlib/BUILD

@@ -130,10 +130,10 @@ cc_library(
 )
 
 tsl_pybind_extension(
-    name = "_extention",
+    name = "_extension",
     testonly = 1,
-    srcs = ["kernel_runner_extention.cc"],
-    visibility = ["//visibility:private"],  # the extention should always be linked via testlib
+    srcs = ["kernel_runner_extension.cc"],
+    visibility = ["//visibility:private"],  # the extension should always be linked via testlib
     deps = [
         ":elemental_kernel_emitter",
         ":kernel_runner",
@@ -160,7 +160,7 @@ pytype_strict_library(
     ],
     srcs_version = "PY3",
     deps = [
-        ":_extention",
+        ":_extension",
         "//xla/codegen/testlib",  # buildcleaner: keep
     ],
 )

xla/backends/cpu/testlib/__init__.py

@@ -14,11 +14,11 @@
 # ==============================================================================
 """Public API for cpu codegen testlib."""
 
-from xla.backends.cpu.testlib import _extention
+from xla.backends.cpu.testlib import _extension
 
 # go/keep-sorted start
-ElementalKernelEmitter = _extention.ElementalKernelEmitter
-KernelRunner = _extention.KernelRunner
-LlvmIrKernelEmitter = _extention.LlvmIrKernelEmitter
-LlvmIrKernelSpec = _extention.LlvmIrKernelSpec
+ElementalKernelEmitter = _extension.ElementalKernelEmitter
+KernelRunner = _extension.KernelRunner
+LlvmIrKernelEmitter = _extension.LlvmIrKernelEmitter
+LlvmIrKernelSpec = _extension.LlvmIrKernelSpec
 # go/keep-sorted end

xla/backends/cpu/testlib/kernel_runner_extention.cc renamed to xla/backends/cpu/testlib/kernel_runner_extension.cc

@@ -60,7 +60,7 @@ void ImportBaseClasses(const nb::module_& kernel_runner_module) {
   nb::module_::import_(absl::StrCat(xla_module, ".codegen.testlib").c_str());
 }
 
-NB_MODULE(_extention, kernel_runner_module) {
+NB_MODULE(_extension, kernel_runner_module) {
   // We depend on the base classes so must import them before python tries to
   // register the derived versions.
   ImportBaseClasses(kernel_runner_module);

xla/codegen/testlib/BUILD

@@ -33,10 +33,10 @@ cc_library(
 )
 
 tsl_pybind_extension(
-    name = "_extention",
+    name = "_extension",
     testonly = 1,
-    srcs = ["kernel_runner_extention.cc"],
-    visibility = ["//visibility:private"],  # the extention should always be linked via testlib
+    srcs = ["kernel_runner_extension.cc"],
+    visibility = ["//visibility:private"],  # the extension should always be linked via testlib
     deps = [
         ":kernel_runner",
         "//xla:comparison_util",
@@ -66,7 +66,7 @@ pytype_strict_library(
     ],
     srcs_version = "PY3",
     deps = [
-        ":_extention",
+        ":_extension",
         "//third_party/py/numpy",
         "//xla/python:xla_extension",
     ],
@@ -82,7 +82,7 @@ py_strict_test(
         "no_oss",
     ],
     deps = [
-        ":_extention",
+        ":_extension",
         ":testlib",
         "//third_party/py/numpy",
         "@absl_py//absl/testing:absltest",

xla/codegen/testlib/__init__.py

@@ -14,14 +14,14 @@
 # ==============================================================================
 """Public API for codegen testlib."""
 
-from xla.codegen.testlib import _extention
+from xla.codegen.testlib import _extension
 
 # Classes
 # go/keep-sorted start
-ComparisonDirection = _extention.ComparisonDirection
-HloInstruction = _extention.HloInstruction
-HloOpcode = _extention.HloOpcode
-KernelEmmitter = _extention.KernelEmitter
-KernelRunner = _extention.KernelRunner
-KernelSpec = _extention.KernelSpec
+ComparisonDirection = _extension.ComparisonDirection
+HloInstruction = _extension.HloInstruction
+HloOpcode = _extension.HloOpcode
+KernelEmmitter = _extension.KernelEmitter
+KernelRunner = _extension.KernelRunner
+KernelSpec = _extension.KernelSpec
 # go/keep-sorted end

xla/codegen/testlib/kernel_runner_extention.cc renamed to xla/codegen/testlib/kernel_runner_extension.cc

@@ -115,7 +115,7 @@ class DummyAddKernelRunner final : public KernelRunner {
 
 }  // namespace
 
-NB_MODULE(_extention, kernel_runner_module) {
+NB_MODULE(_extension, kernel_runner_module) {
   namespace nb = nanobind;
 
   nb::class_<KernelSource>(kernel_runner_module, "KernelSource");

xla/codegen/testlib/kernel_runner_test.py

@@ -15,7 +15,7 @@
 
 from absl.testing import absltest
 import numpy as np
-from xla.codegen.testlib import _extention
+from xla.codegen.testlib import _extension
 from xla.codegen.testlib import utilities as testlib_utilities
 
 
@@ -33,7 +33,7 @@ def test_output_same_as_input(self):
 class DummyKernelRunnerTest(absltest.TestCase):
 
   def test_dummy_kernel(self):
-    runner = _extention.DummyAddKernelRunner()
+    runner = _extension.DummyAddKernelRunner()
     in_arg1 = create_literal(np.array([1, 2, 3, 4], dtype=np.int32))
     in_arg2 = create_literal(np.array([5, 6, 7, 8], dtype=np.int32))
     out_arg = create_literal(np.array([0, 0, 0, 0], dtype=np.int32))

xla/codegen/testlib/utilities.py

@@ -16,7 +16,7 @@
 
 import numpy as np
 
-from xla.codegen.testlib import _extention
+from xla.codegen.testlib import _extension
 from xla.python import xla_extension
 
 
@@ -27,4 +27,4 @@ def create_literal_from_np(array: np.ndarray) -> xla_extension.Literal:
   return literal
 
 # Intentionally rexport-ed to be avalable in the public API.
-opcode_arity = _extention.opcode_arity
+opcode_arity = _extension.opcode_arity