Fix typo and missing header (#1266)

This commit is contained in:
Feng Shijie 2024-07-15 23:20:24 +08:00 committed by GitHub
parent 8c01a7893b
commit 987785d8d7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 1 deletions

View File

@ -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());

View File

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