Fix typo and missing header (ml-explore#1266)

marcosncosta1 · Jul 15, 2024 · 987785d · 987785d
1 parent 8c01a78
commit 987785d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/mlx/primitives.cpp b/mlx/primitives.cpp
@@ -124,7 +124,7 @@ std::vector<array> Primitive::vjp(
     const std::vector<int>&,
     const std::vector<array>&) {
   std::ostringstream msg;
-  msg << "[Primitive::vip] Not implemented for ";
+  msg << "[Primitive::vjp] Not implemented for ";
   print(msg);
   msg << ".";
   throw std::invalid_argument(msg.str());

diff --git a/mlx/random.h b/mlx/random.h
@@ -7,6 +7,7 @@
 
 #include "mlx/array.h"
 #include "mlx/stream.h"
+#include "mlx/utils.h"
 
 namespace mlx::core::random {