Update lazy_select_vs_sort.exs

elixir-nx · Dec 11, 2023 · 0a84aff · 0a84aff
1 parent a202a76
commit 0a84aff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/benchmarks/lazy_select_vs_sort.exs b/benchmarks/lazy_select_vs_sort.exs
@@ -4,7 +4,7 @@ Nx.Defn.global_default_options(compiler: EXLA)
 
 key = Nx.Random.key(System.os_time())
 
-inputs_knn = %{
+inputs_k_smallest = %{
   "10" => elem(Nx.Random.shuffle(key, Nx.iota({10})), 0),
   "100" => elem(Nx.Random.shuffle(key, Nx.iota({100})), 0),
   "1000" => elem(Nx.Random.shuffle(key, Nx.iota({1000})), 0),
@@ -23,7 +23,7 @@ Benchee.run(
   },
   time: 10,
   memory_time: 2,
-  inputs: inputs_knn
+  inputs: inputs_k_smallest
 )