Various typo fixes in exla/

exla/lib/exla/device_buffer.ex

@@ -15,7 +15,7 @@ defmodule EXLA.DeviceBuffer do
   end
 
   @doc """
-  Places the given binary `buffer` on the given `device` using `client`.
+  Places the given binary `data` on the given `device` using `client`.
   """
   def place_on_device(data, %EXLA.Typespec{} = typespec, client = %Client{}, device_id)
       when is_integer(device_id) and is_binary(data) do

exla/lib/exla/mlir/module.ex

@@ -68,7 +68,7 @@ defmodule EXLA.MLIR.Module do
       It defaults to 1 but you can set it if you want to enable single-program
       multiple data.
 
-    * `:use_spmd` - enables Single-Program Multiple-Data partioning.
+    * `:use_spmd` - enables Single-Program Multiple-Data partitioning.
       This is set to true if `:num_partitions` is more than one, otherwise is `false`.
 
     * `:module_compilation` - either `:to_mlir` or `:to_pjrt`. The default is `:to_pjrt`.

exla/test/exla/defn/lock_test.exs

@@ -1,9 +1,9 @@
-defmodule EXLA.Defn.LockeTest do
+defmodule EXLA.Defn.LockTest do
   use ExUnit.Case, async: true
 
   alias EXLA.Defn.Lock, as: L
 
-  test "locks a given key while process is alive", config do
+  test "locks a given key while the process is alive", config do
     parent = self()
 
     task1 =
@@ -96,7 +96,7 @@ defmodule EXLA.Defn.LockeTest do
     assert Task.await(task3)
   end
 
-  test "transfers", config do
+  test "transfers locks correctly", config do
     parent = self()
 
     task1 =

exla/test/exla/defn/vectorize_test.exs

@@ -194,7 +194,7 @@ defmodule EXLA.Defn.VectorizeTest do
     end
 
     test "simple cond" do
-      # this tests the case where we have a two vectorized predicates
+      # This tests the case where we have two vectorized predicates
       pred1 = Nx.vectorize(~VEC[1 0 0], :pred)
       pred2 = Nx.vectorize(~VEC[0 0 0], :pred)